epoch_3_step_8166_rank*.pt