From 9aebc711c96c5d88eaba5fe385370de7924e6ec7 Mon Sep 17 00:00:00 2001 From: tholor Date: Thu, 7 Feb 2019 11:49:38 +0100 Subject: [PATCH] adjust error message related to args.do_eval --- examples/run_lm_finetuning.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/run_lm_finetuning.py b/examples/run_lm_finetuning.py index 0ec2b48823..53bf5e59d7 100644 --- a/examples/run_lm_finetuning.py +++ b/examples/run_lm_finetuning.py @@ -501,8 +501,8 @@ def main(): if n_gpu > 0: torch.cuda.manual_seed_all(args.seed) - if not args.do_train and not args.do_eval: - raise ValueError("At least one of `do_train` or `do_eval` must be True.") + if not args.do_train: + raise ValueError("Training is currently the only implemented execution option. Please set `do_train`.") if os.path.exists(args.output_dir) and os.listdir(args.output_dir): raise ValueError("Output directory ({}) already exists and is not empty.".format(args.output_dir))