These datasets are collected in the same format as robomimic. To train on this dataset (and your own datasets), please modify your config file as follows:
- Change
dataset_path
to the path ofhdf5
file. - Use the following
shape_meta
.
shape_meta:
action:
shape:
- 4
obs:
pos:
shape:
- 4
type: low_dim
camera0:
shape:
- 3
- 120
- 160
type: rgb
camera1:
shape:
- 3
- 120
- 160
type: rgb