2 parents 5d6c360 + a9019db commit 96b6e67Copy full SHA for 96b6e67
1 file changed
TensorFlow/LanguageModeling/BERT/run_pretraining.py
@@ -628,7 +628,8 @@ def main(_):
628
629
if __name__ == "__main__":
630
flags.mark_flag_as_required("input_files_dir")
631
- flags.mark_flag_as_required("eval_files_dir")
+ if FLAGS.do_eval:
632
+ flags.mark_flag_as_required("eval_files_dir")
633
flags.mark_flag_as_required("bert_config_file")
634
flags.mark_flag_as_required("output_dir")
635
if FLAGS.use_xla and FLAGS.manual_fp16:
0 commit comments