This commit is contained in:
Cadene
2024-03-09 14:18:47 +00:00
committed by Simon Alibert
parent a54a0feb63
commit cde866dac0
15 changed files with 34 additions and 0 deletions

BIN
tests/data/pusht/action.memmap LFS Normal file

Binary file not shown.

BIN
tests/data/pusht/episode.memmap LFS Normal file

Binary file not shown.

BIN
tests/data/pusht/frame_id.memmap LFS Normal file

Binary file not shown.

View File

@@ -0,0 +1 @@
{"action": {"device": "cpu", "shape": [50, 2], "dtype": "torch.float32"}, "episode": {"device": "cpu", "shape": [50], "dtype": "torch.int64"}, "frame_id": {"device": "cpu", "shape": [50], "dtype": "torch.int64"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}

BIN
tests/data/pusht/next/done.memmap LFS Normal file

Binary file not shown.

View File

@@ -0,0 +1 @@
{"reward": {"device": "cpu", "shape": [50, 1], "dtype": "torch.float32"}, "done": {"device": "cpu", "shape": [50, 1], "dtype": "torch.bool"}, "success": {"device": "cpu", "shape": [50, 1], "dtype": "torch.bool"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}

Binary file not shown.

View File

@@ -0,0 +1 @@
{"image": {"device": "cpu", "shape": [50, 3, 96, 96], "dtype": "torch.float32"}, "state": {"device": "cpu", "shape": [50, 2], "dtype": "torch.float32"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

View File

@@ -0,0 +1 @@
{"image": {"device": "cpu", "shape": [50, 3, 96, 96], "dtype": "torch.float32"}, "state": {"device": "cpu", "shape": [50, 2], "dtype": "torch.float32"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}

Binary file not shown.

BIN
tests/data/pusht/stats.pth Normal file

Binary file not shown.