Skip to content

Optimizes training progress bar #1352

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

harshav17
Copy link

What this does

Previously progress bar was hard to read. This PR tries to make the PR more visually appealing.

Screenshot 2025-06-18 at 9 19 15 PM

How it was tested

Manually by running the train.py script

python lerobot/scripts/train.py   --policy.path=lerobot/smolvla_base   --dataset.repo_id=harshav17/salt-test   --batch_size=48   --steps=20000   --output_dir=outputs/train/my_smolvla   --job_name=my_smolvla_training   --policy.device=cuda   --wandb.enable=true

How to checkout & try? (for the reviewer)

Simply run the following script to train smolvla_base model.

python lerobot/scripts/train.py   --policy.path=lerobot/smolvla_base   --dataset.repo_id=harshav17/salt-test   --batch_size=48   --steps=20000   --output_dir=outputs/train/my_smolvla   --job_name=my_smolvla_training   --policy.device=cuda   --wandb.enable=true

SECTION TO REMOVE BEFORE SUBMITTING YOUR PR

Note: Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR. Try to avoid tagging more than 3 people.

Note: Before submitting this PR, please read the contributor guideline.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant