Merge pull request #299 from marian-code/patch-1

fix error in definition of numb_test possible type
deepmodeling · Nov 26, 2020 · b96112e · b96112e
2 parents bf1f001 + e90d33d
commit b96112e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/train/argcheck.py b/source/train/argcheck.py
@@ -321,7 +321,7 @@ def training_args():
         Argument("seed", [int,None], optional = True, doc = doc_seed),
         Argument("disp_file", str, optional = True, default = 'lcueve.out', doc = doc_disp_file),
         Argument("disp_freq", int, optional = True, default = 1000, doc = doc_disp_freq),
-        Argument("numb_test", int, optional = True, default = 1, doc = doc_numb_test),
+        Argument("numb_test", [list,int,str], optional = True, default = 1, doc = doc_numb_test),
         Argument("save_freq", int, optional = True, default = 1000, doc = doc_save_freq),
         Argument("save_ckpt", str, optional = True, default = 'model.ckpt', doc = doc_save_ckpt),
         Argument("disp_training", bool, optional = True, default = True, doc = doc_disp_training),