train_state.json

{"current_epoch": 17, "current_step": 1001}