From a40d0cbcc79447af6ac9c3bd666150f998dcfc38 Mon Sep 17 00:00:00 2001 From: Thomas Wolf Date: Sat, 8 Jun 2024 12:16:33 +0200 Subject: [PATCH] Update README.md Co-authored-by: Remi --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 947983e2..ef8ea786 100644 --- a/README.md +++ b/README.md @@ -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, ...}