6.2 ACT 训练

⚠️ 训练需要 NVIDIA GPU，无法在树莓派或 Jetson Nano 上运行。

训练日志示例：

INFO step:100 smpl:800 ep:3 loss:4.547 grdn:131.117 lr:1.0e-05

参考用时：10 组数据，3090 约 2 小时。

ACT 训练

linux: lerobot-train \
    --dataset.repo_id=${HF_USER}/so101_test \
    --policy.type=act \
    --output_dir=outputs/train/act_so101_test \
    --job_name=act_so101_test \
    --policy.device=cuda \
    --policy.push_to_hub=false \
    --wandb.enable=false
macos: lerobot-train \
    --dataset.repo_id=${HF_USER}/so101_test \
    --policy.type=act \
    --output_dir=outputs/train/act_so101_test \
    --job_name=act_so101_test \
    --policy.device=cuda \
    --policy.push_to_hub=false \
    --wandb.enable=false
windows: lerobot-train `
    --dataset.repo_id=${HF_USER}/so101_test `
    --policy.type=act `
    --output_dir=outputs/train/act_so101_test `
    --job_name=act_so101_test `
    --policy.device=cuda `
    --policy.push_to_hub=false `
    --wandb.enable=false

恢复中断的训练

linux: lerobot-train \
    --config_path=outputs/train/act_so101_test/checkpoints/last/pretrained_model/train_config.json \
    --resume=true
macos: lerobot-train \
    --config_path=outputs/train/act_so101_test/checkpoints/last/pretrained_model/train_config.json \
    --resume=true
windows: lerobot-train `
    --config_path=outputs/train/act_so101_test/checkpoints/last/pretrained_model/train_config.json `
    --resume=true