evaluation bug fixed

2020-04-30 02:00:10 -07:00 · 2020-04-30 02:00:10 -07:00 · 77ad950ba1
parent fa0ef5b687
commit 77ad950ba1
1 changed files with 1 additions and 1 deletions
--- a/genienlp/util.py
+++ b/genienlp/util.py
@ -445,7 +445,7 @@ def load_config_json(args):
        # train and predict scripts have these arguments in common. We use the values from train only if they are not provided in predict
        overwrite = ['val_batch_size', 'num_beams']
        for o in overwrite:
-            if getattr(args, o) is None:
+            if o not in args or args[o] is None:
                retrieve.append(o)

        for r in retrieve: