Skip to content

Commit

Permalink
step start from 1
Browse files Browse the repository at this point in the history
  • Loading branch information
kamalkraj committed Nov 27, 2021
1 parent 04683c0 commit 2b87fda
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion examples/flax/token-classification/run_flax_ner.py
Original file line number Diff line number Diff line change
Expand Up @@ -593,7 +593,8 @@ def compute_metrics():
total=step_per_epoch,
desc="Training...",
position=1,
)
),
start=1,
):
state, train_metric, dropout_rngs = p_train_step(state, batch, dropout_rngs)
train_metrics.append(train_metric)
Expand Down

0 comments on commit 2b87fda

Please sign in to comment.