Downloads · 30 days
4
13% of all-time downloads
staudi25/xvla_lego_pickup_mega_plus
xvla_lego_pickup_mega_plus is a machine learning model from staudi25. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
lerobot-train --dataset.repoid=staudi25/legopickupmegaplus --jobname=xvlaso101legopickup --policy.path=staudi25/xvlalegopickupmega --policy.device=cuda --wandb.enable=false --dataset.imagetransforms.enable=true --logf…
Downloads · 30 days
4
13% of all-time downloads
All-time downloads
32
Public
Parameters
880M
1.8 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors1.8 GB · 100%
From the Hugging Face model README
lerobot-train --dataset.repo_id=staudi25/lego_pickup_mega_plus --job_name=xvla_so101_lego_pickup --policy.path=staudi25/xvla_lego_pickup_mega --policy.device=cuda --wandb.enable=false --dataset.image_transforms.enable=true --log_freq=1 --steps=20000 --save_freq=5000 --policy.repo_id=staudi25/xvla_lego_pickup_mega_plus --policy.action_mode=auto --policy.freeze_vision_encoder=false --policy.freeze_language_encoder=false --policy.train_policy_transformer=true --policy.train_soft_prompts=true --policy.dtype=bfloat16 --output_dir=outputs/train/xvla_20k_lego_pickup_mega_plus --rename_map='{"observation.images.desk_cam": "observation.images.image", "observation.images.gripper": "observation.images.image2"}'