Safetensors

Train script: https://github.com/huggingface/lerobot/blob/8e2a39444255d62869d2fd63ea4f8a157236a029/lerobot/scripts/train.py

Train command: python /workspace/.venv/lib/python3.13/site-packages/lerobot/scripts/train.py --dataset.repo_id=${repo_id} --dataset.root=${repo_id} --policy.type=diffusion --output_dir=outputs/train/${job_name} --job_name=${job_name} --policy.device=cuda --wandb.enable=true

Wandb: https://wandb.ai/vsaucemichaelhere00-university-of-stuttgart/lerobot/runs/c7n113qa/workspace?nw=nwuservsaucemichaelhere00

Loss:

image/png

Downloads last month
2
Safetensors
Model size
0.3B params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Dataset used to train domhel/studytable_open_drawer_depth_aug