wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
544 Commits 131 Branches 0 Tags 244 MiB
Commit Graph

3 Commits

Author SHA1 Message Date
Radek Osmulski 621f69d98f Implement 2nd round of review changes 2024-05-30 07:36:58 +10:00
Radek Osmulski 566a8aa98e Implement review feedback 2024-05-30 07:36:58 +10:00
Radek Osmulski e7abcc2ffd add EpisodeAwareSampler 
EpisodeAwareSampler allows the dropping of first n or last n frames
from each episode.

Additionally, you can select which episodes to run on by passig in
their ids (complements functionality of the hf dataset slice syntax
to give extra flexibility to the user).

* modify train.py and diffusion.yaml to leverage new functionality
  and train on pusht with last 7 steps per episode dropped
* add unit tests
 2024-05-30 07:36:58 +10:00