train_state.json

{"current_epoch": 21, "current_step": 1428}