Properly use steps_per_epoch and num_epoch in BERT classify benchmark (#6985)
Properly use steps_per_epoch and num_epoch in BERT classify benchmark (#6985)

* move step and epoch counts after super init call

* move comment block

* move super to the top

tombstone/models,alexgorban/models,tombstone/models,alexgorban/models,tombstone/models,alexgorban/models,alexgorban/models,tombstone/models,alexgorban/models,tombstone/models,tombstone/models