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

* step start from 1

* Updated cur_step calcualtion
huggingface/transformers,huggingface/transformers,huggingface/transformers,huggingface/transformers