how to finetune this smolvla with my own dataset?
#1
by
Xiaoyan97
- opened
Has your model worked? In my attempt, I recorded 40 action records at the same position and trained 20,000 steps using the official fine-tuning code. However, my model was unable to complete the actions.
python lerobot/scripts/train.py --policy.path=lerobot/smolvla_base --dataset.repo_id=Xiaoyan97/pick_up_scissors_v1 --dataset.root=Xiaoyan97/pick_up_scissors_v1 --batch_size=64 --steps=20000 --output_dir=outputs/smolvla_v1 --job_name=my_smolvla_training --policy.device=cuda
Is it that my training steps are insufficient or for some other reason?
This one has not worked as well for me. I may remove this particular example. I think I was doing too much between each episode.