Update README.md

Co-authored-by: Remi <re.cadene@gmail.com>
This commit is contained in:
Thomas Wolf 2024-06-08 12:16:33 +02:00 committed by GitHub
parent 3717639f11
commit a40d0cbcc7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -165,7 +165,7 @@ dataset attributes:
│ ├ next.done: bool: indicates the end of en episode ; True for the last frame in each episode
│ └ index: int64: general index in the whole dataset
├ episode_data_index: contains 2 tensors with the start and end indices of each episode
│ ├ from: 1D int64 tensor of first frame index for each episode: shape (num episodes,) start with 0
│ ├ from: 1D int64 tensor of first frame index for each episode: shape (num episodes,) starts with 0
│ └ to: 1D int64 tensor of last frame index for each episode: shape (num episodes,)
├ stats: a dictionary of statistics (max, mean, min, std) for each feature in the dataset, for instance
│ ├ observation.images.cam_high: {'max': tensor of same shape as the observation.images.cam_high feature, ...}