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)
|