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

[Flax] token-classification model steps enumerate start from 1 (#14547)

Browse Source 
* step start from 1

* Updated cur_step calcualtion
This commit is contained in:
Kamal Raj
2021-11-29 21:55:59 +05:30
committed by GitHub
parent cea17acd8c
commit 2bd950ca47
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/flax/token-classification/run_flax_ner.py
View File

@@ -598,7 +598,7 @@ def main():
state, train_metric, dropout_rngs = p_train_step(state, batch, dropout_rngs) state, train_metric, dropout_rngs = p_train_step(state, batch, dropout_rngs)
train_metrics.append(train_metric) train_metrics.append(train_metric)
cur_step = epoch * step_per_epoch + step cur_step = (epoch * step_per_epoch) + (step + 1)
if cur_step % training_args.logging_steps == 0 and cur_step > 0: if cur_step % training_args.logging_steps == 0 and cur_step > 0:
# Save metrics # Save metrics