SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[Flax] Correct logging steps flax (#12515)

Browse Source 
* fix_torch_device_generate_test

* remove @

* push
This commit is contained in:
Patrick von Platen
2021-07-05 18:21:00 +01:00
committed by GitHub
parent bb4ac2b5a8
commit d0f7508abe
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -574,7 +574,7 @@ def main():
cur_step = epoch * (len(train_dataset) // train_batch_size) + step cur_step = epoch * (len(train_dataset) // train_batch_size) + step
if cur_step % training_args.logging_steps and cur_step > 0: if cur_step % training_args.logging_steps == 0 and cur_step > 0:
# Save metrics # Save metrics
train_metric = unreplicate(train_metric) train_metric = unreplicate(train_metric)
train_time += time.time() - train_start train_time += time.time() - train_start

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

@@ -608,7 +608,7 @@ if __name__ == "__main__":
cur_step = epoch * num_train_samples + step cur_step = epoch * num_train_samples + step
if cur_step % training_args.logging_steps and cur_step > 0: if cur_step % training_args.logging_steps == 0 and cur_step > 0:
# Save metrics # Save metrics
train_metric = jax_utils.unreplicate(train_metric) train_metric = jax_utils.unreplicate(train_metric)
train_time += time.time() - train_start train_time += time.time() - train_start

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

@@ -724,7 +724,7 @@ if __name__ == "__main__":
cur_step = epoch * num_train_samples + step cur_step = epoch * num_train_samples + step
if cur_step % training_args.logging_steps and cur_step > 0: if cur_step % training_args.logging_steps == 0 and cur_step > 0:
# Save metrics # Save metrics
train_metric = jax_utils.unreplicate(train_metric) train_metric = jax_utils.unreplicate(train_metric)
train_time += time.time() - train_start train_time += time.time() - train_start