boris commited on
Commit
900136f
1 Parent(s): 0a0080b

fix: eval_steps belongs to training_args

Browse files
Files changed (1) hide show
  1. seq2seq/run_seq2seq_flax.py +1 -1
seq2seq/run_seq2seq_flax.py CHANGED
@@ -790,7 +790,7 @@ def main():
790
  # log metrics
791
  wandb_log(unreplicate(train_metric), step=global_step, prefix='tran')
792
 
793
- if global_step % data_args.eval_steps == 0:
794
  run_evaluation()
795
 
796
  train_time += time.time() - train_start
 
790
  # log metrics
791
  wandb_log(unreplicate(train_metric), step=global_step, prefix='tran')
792
 
793
+ if global_step % training_args.eval_steps == 0:
794
  run_evaluation()
795
 
796
  train_time += time.time() - train_start