train_state.json

{"current_epoch": 22, "current_step": 1496}