iFaz's picture
Add auto-generated model card from override dicts
868cecc verified
metadata
datasets:
  - lerobot/aloha_sim_transfer_cube_human
language: en
library_name: lerobot
license: apache-2.0
tags:
  - robotics
  - imitation-learning
  - diffusion
  - mujoco
  - pytorch_model_hub_mixin

DIFFUSION Policy β€” aloha_diff_baseline

Trained with LeRobot.
Date: 2026-05-04 17:20
Policy type: diffusion | Device: cuda


πŸ“¦ Dataset

Parameter Value
dataset.repo_id lerobot/aloha_sim_transfer_cube_human

πŸ‹οΈ Training Config

Parameter Value
steps 5000
batch_size 8
eval_freq 500
save_freq 1000
num_workers 4
seed 42
eval.n_episodes 10
eval.batch_size 10

πŸ“ Policy Architecture

All defaults β€” no overrides applied.


🎯 Eval Config

Parameter Value
env.type aloha
env.task AlohaTransferCube-v0
eval.n_episodes 5
eval.batch_size 1
policy.path /kaggle/working/outputs/train/aloha_act_baseline/checkpoints/last/pretrained_model

πŸ“Š Eval Results

Metric Value
Episodes 5
Success rate 0.0%
Avg sum reward 0.00
Avg max reward 0.00
Eval time (s) 1126.8

Citation

@misc{cadene2024lerobot,
  author = {Cadene, Remi and Alibert, Simon and others},
  title  = {LeRobot},
  year   = {2024},
  url    = {https://github.com/huggingface/lerobot}
}