Spaces:
Running
Running
fix: eval_steps belongs to training_args
Browse files
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 %
|
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
|