riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-04 21:30:07 +06:00
Code Issues Actions 28 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats 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