From: Francois Fleuret <francois@fleuret.org>
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) + ' ###############')