Spaces:
Running
Running
feat: hardcode eval_steps
Browse files
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)
|