Views
No views yet
checkpoints/stepNNN.ckpt/pytorch_model.pt = consolidated fp32 state_dict..hydra/config.yaml = the training config — required to rebuild the model architecture for eval.| file | step |
|---|---|
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 |
1import torch
2sd = torch.load("checkpoints/step500.ckpt/pytorch_model.pt", map_location="cpu")Note:rollouts.jsonl(RL training telemetry) is not included.