stage2 latest raw checkpoints: step_150000, step_160000