From e90d33db1e82b150839d300678efe5e47c2f0632 Mon Sep 17 00:00:00 2001 From: marian-code Date: Tue, 24 Nov 2020 13:34:45 +0100 Subject: [PATCH] fix error in definition of numb_test possible type numb_test can be input as str, int, or list not only list --- source/train/argcheck.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/source/train/argcheck.py b/source/train/argcheck.py index 89034180ba..d87d41857c 100644 --- 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),