This commit is contained in:
Sina 2020-04-30 16:08:11 -07:00
parent 77ad950ba1
commit 517f87b66b
1 changed files with 1 additions and 1 deletions

View File

@ -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 o not in args or args[o] is None:
if o not in args or getattr(args, o) is None:
retrieve.append(o)
for r in retrieve: