Skip to content
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

Adds a script for recording training data from a Gym simulation and replaying in sim #437

Open
wants to merge 28 commits into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from 18 commits
Commits
Show all changes
28 commits
Select commit Hold shift + click to select a range
e8c214d
added data recording in sim example
samzapo Sep 11, 2024
5ad1bcf
Cleaned up for PR
samzapo Sep 12, 2024
6a66f4a
checkpoint
samzapo Sep 12, 2024
59270e0
rolled back changes that recorded reset observations.
samzapo Sep 12, 2024
735d5e5
Merge branch 'main' of github.com:huggingface/lerobot into record_gym…
samzapo Sep 12, 2024
38f9abd
first commit
samzapo Sep 13, 2024
29326a6
checkpoint, code in place for running sim and saving data
samzapo Sep 13, 2024
a77c005
checkpoint, setting-up replaying
samzapo Sep 13, 2024
9cce7d7
checkpoint
samzapo Sep 16, 2024
30baf69
working script lerobot/scripts/record_and_replay_teleop_dataset.py. …
samzapo Sep 16, 2024
66d785c
Checkpoint working replay from disk
samzapo Sep 16, 2024
a3100bd
Cleaned up code (removed runner class). Replay fails during compute …
samzapo Sep 17, 2024
125a438
Cleaned up code (removed 'runner' class). Replay fails during comput…
samzapo Sep 17, 2024
703ee07
Fixed gripper open/close issue
samzapo Sep 17, 2024
525b2f9
dummy code for teleop using leader arm
samzapo Sep 17, 2024
1a1ff0d
Randomizer re-seeds before each sim run
samzapo Sep 18, 2024
067ab01
Merge remote-tracking branch 'upstream/main' into record_gym_script
samzapo Sep 19, 2024
321cb62
does not double reset on start, corrected teleop docs
samzapo Sep 20, 2024
2c29606
Merge remote-tracking branch 'upstream/main' into record_gym_script
samzapo Sep 30, 2024
a47a08b
checkpoint: reading joint angles from leader arm
samzapo Oct 1, 2024
afa633b
Teleoperating with leader arm in sim
samzapo Oct 2, 2024
139dc46
(checkpoint) adding kwargs config
samzapo Oct 2, 2024
1731e35
Using kwargs config instead of explicitly setting args in environment…
samzapo Oct 3, 2024
ce8fe1b
Updated version of gym-drake-lca that uses IDYN controller, and bette…
samzapo Oct 3, 2024
862f40d
Merge remote-tracking branch 'upstream/main' into record_gym_script
samzapo Oct 3, 2024
ee66983
removed explicitly requesting vcodec
samzapo Oct 4, 2024
21bd0e0
checkpoint: final update
samzapo Oct 10, 2024
666e2b7
Merge branch 'main' of github.com:huggingface/lerobot into record_gym…
samzapo Oct 10, 2024
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions lerobot/common/datasets/lerobot_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -126,8 +126,8 @@ def tolerance_s(self) -> float:
are not close enough from the requested frames. It is only used when `delta_timestamps`
is provided or when loading video frames from mp4 files.
"""
# 1e-4 to account for possible numerical error
return 1 / self.fps - 1e-4
# Add 1e-4 seconds to expected step size to account for possible numerical error
return 1 / self.fps + 1e-4

def __len__(self):
return self.num_samples
Expand Down Expand Up @@ -357,8 +357,8 @@ def tolerance_s(self) -> float:
are not close enough from the requested frames. It is only used when `delta_timestamps`
is provided or when loading video frames from mp4 files.
"""
# 1e-4 to account for possible numerical error
return 1 / self.fps - 1e-4
# Add 1e-4 seconds to expected step size to account for possible numerical error
return 1 / self.fps + 1e-4

def __len__(self):
return self.num_samples
Expand Down
Loading
Loading