Skip to content

lerobot training colab code is broken, below is the corrected code #603

@therealadityashankar

Description

@therealadityashankar

So in the training colab at https://colab.research.google.com/github/huggingface/notebooks/blob/main/lerobot/training-act.ipynb

This is the corrected code, the training script broke, due to the lerobot github repository reorganizing itself

!cd lerobot && python lerobot/scripts/train.py \ --dataset.repo_id=${HF_USER}/il_gym0 \ --policy.type=act \ --output_dir=outputs/train/il_sim_test0 \ --job_name=il_sim_test \ --policy.device=cuda \ --wandb.enable=true 

should instead be

!python lerobot/src/lerobot/scripts/train.py \ --dataset.repo_id=${HF_USER}/jenga_dataset_2_video \ --policy.type=act \ --output_dir=outputs/train/il_sim_test0 \ --job_name=il_sim_test \ --policy.device=cuda \ --policy.push_to_hub=false 

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions