dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[Flax] Fix another bug in logging steps (#12516) 

* fix_torch_device_generate_test

* remove @

* up
This commit is contained in:
Patrick von Platen 2021-07-05 18:35:22 +01:00 committed by GitHub
parent d0f7508abe
commit 4605b2b8ec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/flax/language-modeling/run_mlm_flax.py
View File

@ -606,7 +606,7 @@ if __name__ == "__main__":
state, train_metric, dropout_rngs = p_train_step(state, model_inputs, dropout_rngs)
train_metrics.append(train_metric)
cur_step = epoch * num_train_samples + step
cur_step = epoch * (num_train_samples // train_batch_size) + step
if cur_step % training_args.logging_steps == 0 and cur_step > 0:
# Save metrics

2
examples/flax/language-modeling/run_t5_mlm_flax.py
View File

@ -722,7 +722,7 @@ if __name__ == "__main__":
state, train_metric, dropout_rngs = p_train_step(state, model_inputs, dropout_rngs)
train_metrics.append(train_metric)
cur_step = epoch * num_train_samples + step
cur_step = epoch * (num_train_samples // train_batch_size) + step
if cur_step % training_args.logging_steps == 0 and cur_step > 0:
# Save metrics