From: Francois Fleuret Date: Sun, 18 Jun 2017 10:31:55 +0000 (+0200) Subject: Added --problems. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=c3182d1650b6c91be5f012885725b49eacd68db2;p=pysvrt.git Added --problems. --- diff --git a/cnn-svrt.py b/cnn-svrt.py index 704b003..7fde85d 100755 --- a/cnn-svrt.py +++ b/cnn-svrt.py @@ -77,6 +77,10 @@ parser.add_argument('--test_loaded_models', type = distutils.util.strtobool, default = 'False', help = 'Should we compute the test errors of loaded models') +parser.add_argument('--problems', + type = str, default = '1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23', + help = 'What problem to process') + args = parser.parse_args() ###################################################################### @@ -276,7 +280,7 @@ else: log_string('using_uncompressed_vignettes') VignetteSet = svrtset.VignetteSet -for problem_number in range(1, 24): +for problem_number in map(int, args.problems.split(',')): log_string('############### problem ' + str(problem_number) + ' ###############')