Views
No views yet
1CUDA_VISIBLE_DEVICES=0 lerobot-train \
2 --dataset.repo_id=qiaoceng/AIC-data_augment-v2 \
3 --dataset.image_transforms.enable=true \
4 --policy.type=act \
5 --output_dir=/mnt/SSD7/college_student/outputs/train/act-v3 \
6 --job_name=cupstacking \
7 --policy.device=cuda \
8 --wandb.enable=true \
9 --policy.repo_id=qiaoceng/AIC-act-v3-100000 \
10 --policy.chunk_size=50 \
11 --policy.n_action_steps=50 \
12 --batch_size=32 \
13 --steps=100000 \
14 --save_freq=20000