train_state.json

{"current_epoch": 16, "current_step": 858}