boris commited on
Commit
283adc6
1 Parent(s): 754f876

fix: correct arg

Browse files
Files changed (1) hide show
  1. seq2seq/run_seq2seq_flax.py +1 -1
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 % training_args.save_model_steps == 0:
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