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
|