fix(g05): honor autoregressive loss weights

This commit is contained in:
Pepijn
2026-07-30 12:28:40 +02:00
parent eb9a4dae79
commit 65a8b40484
3 changed files with 131 additions and 6 deletions
+3
View File
@@ -151,6 +151,9 @@ G0.5 implements LeRobot's training surface natively: `forward` computes
assistant-token cross entropy and flow-matching loss, the policy exposes
VLM/vision/action optimizer groups, and the checkpoint can be saved, resumed,
and loaded by the normal LeRobot scripts.
The objective preserves the packaged `ar.ce_weight`, optional
`ar.ce_z_loss_scale`, and `fm.fm_weight`; the autoregressive loss jointly covers
CoT text and ActionCodec targets.
For example, fine-tune the private SO-101 checkpoint on a LeRobot dataset:
### Training Command Example