checkpoint-2310/rng_state.pth