train_state.json

{"current_epoch": 20, "current_step": 1360}