GR00T SO101 Ball In Box

Training

• Base model: nvidia/GR00T-N1.5-3B
• Dataset: yeopeter1031/so101-ball-in-box
• Policy type: groot
• Steps: 2500
• Batch size: 16
• Effective samples: 40000
• Image transforms: enabled
• Final checkpoint: 002500
• Final logged loss: 0.065

Image Transforms

• Color jitter: brightness, contrast, saturation, hue
• Sharpness jitter
• Random affine

Files

• model.safetensors: fine-tuned policy weights
• config.json: policy config
• train_config.json: training config
• policy_preprocessor.json: input preprocessing config
• policy_postprocessor.json: output postprocessing config

Notes