boris commited on
Commit
4c5e5a7
1 Parent(s): 900136f

feat: hardcode eval_steps

Browse files
Files changed (1) hide show
  1. seq2seq/run_seq2seq_flax.py +3 -0
seq2seq/run_seq2seq_flax.py CHANGED
@@ -357,6 +357,9 @@ def main():
357
  else:
358
  model_args, data_args, training_args = parser.parse_args_into_dataclasses()
359
 
 
 
 
360
  if (
361
  os.path.exists(training_args.output_dir)
362
  and os.listdir(training_args.output_dir)
 
357
  else:
358
  model_args, data_args, training_args = parser.parse_args_into_dataclasses()
359
 
360
+ logger.warning(f"eval_steps has been manually hardcoded") # TODO: remove it later, convenient for now
361
+ training_args.eval_steps = 400
362
+
363
  if (
364
  os.path.exists(training_args.output_dir)
365
  and os.listdir(training_args.output_dir)