grpo_stable_reasoning_nolow_0711
GRPO RL checkpoints for a SimLingo-style driving VLA (InternVL2-1B + continuous speed-waypoint head).
- Training script:
run_grpo_stable_reasoning_nolow (0711) - Each
checkpoints/stepNNN.ckpt/pytorch_model.pt= consolidated fp32state_dict. .hydra/config.yaml= the training config (needed to rebuild the model for eval).
| file | step |
|---|---|
checkpoints/step100.ckpt/pytorch_model.pt |
100 |
checkpoints/step200.ckpt/pytorch_model.pt |
200 |
checkpoints/step300.ckpt/pytorch_model.pt |
300 |
checkpoints/step400.ckpt/pytorch_model.pt |
400 |
checkpoints/step500.ckpt/pytorch_model.pt |
500 |
import torch
sd = torch.load("checkpoints/step300.ckpt/pytorch_model.pt", map_location="cpu")