remove __name__ outside script

This commit is contained in:
Cadene 2024-04-10 17:16:44 +00:00
parent f8c5a2eb10
commit 3914831585
1 changed files with 0 additions and 13 deletions

View File

@ -209,16 +209,3 @@ class PushtDataset(torch.utils.data.Dataset):
self.data_dict[key] = torch.cat([x[key] for x in ep_dicts])
self.data_dict["index"] = torch.arange(0, total_frames, 1)
if __name__ == "__main__":
dataset = PushtDataset(
"pusht",
root=Path("data"),
delta_timestamps={
"observation.image": [0, -1, -0.2, -0.1],
"observation.state": [0, -1, -0.2, -0.1],
"action": [-0.1, 0, 1, 2, 3],
},
)
dataset[10]