Spaces:
Running
Running
fix: correct arg
Browse files
seq2seq/run_seq2seq_flax.py
CHANGED
@@ -802,7 +802,7 @@ def main():
|
|
802 |
if global_step % training_args.eval_steps == 0:
|
803 |
run_evaluation()
|
804 |
|
805 |
-
if global_step %
|
806 |
run_save_model(global_step, epoch)
|
807 |
|
808 |
# log final train metrics
|
|
|
802 |
if global_step % training_args.eval_steps == 0:
|
803 |
run_evaluation()
|
804 |
|
805 |
+
if global_step % data_args.save_model_steps == 0:
|
806 |
run_save_model(global_step, epoch)
|
807 |
|
808 |
# log final train metrics
|