Views
No views yet
video.image, video.wrist_image) plus an 8-D state and predicts 16 joint-space actions using the diffusion head.nvidia/GR00T-N1.5-3Bdenoising_steps=4. Metrics are on unnormalized actions.| Metric | Value |
|---|---|
| Average MSE | 0.03396 |
| Median MSE | 0.03179 |
| Std MSE | 0.01547 |
| Max MSE | 0.06431 |
| Fraction ≤ 0.05 | 80.0% |
| Fraction ≤ 0.075 | 100.0% |
| Fraction ≤ 0.10 | 100.0% |
1from gr00t.experiment.data_config import load_data_config
2from gr00t.model.policy import Gr00tPolicy
3
4ckpt = 'Tacoin/GR00T-N1.5-3B-LIBERO-OBJECT-8K'
5data_config = load_data_config('libero_gr00t')
6policy = Gr00tPolicy(
7 model_path=ckpt,
8 modality_config=data_config.modality_config(),
9 modality_transform=data_config.transform(),
10 embodiment_tag='new_embodiment',
11 denoising_steps=4,
12)policy.get_action(...) to obtain the 16-step plan.| Path | Description |
|---|---|
config.json | Transformer config for the action head. |
model-0000x-of-00002.safetensors | Sharded weights. |
model.safetensors.index.json | Weight shard index. |
experiment_cfg/metadata.json | Dataset statistics for normalization. |
optimizer.pt, scheduler.pt, rng_state.pth | Optimizer state for resuming. |
trainer_state.json | Trainer snapshot (loss curves, etc.). |