robomimic package#

Subpackages#

Module contents#

Record all dataset links in this function.

Record all dataset links in this function.

Helper function to register dataset link in global dictionary. Also takes a @horizon parameter - this corresponds to the evaluation rollout horizon that should be used during training.

Parameters
  • task (str) – name of task for this dataset

  • dataset_type (str) – type of dataset (usually identifies the dataset source)

  • hdf5_type (str) – type of hdf5 - usually one of “raw”, “low_dim”, or “image”, to identify the kind of observations in the dataset

  • link (str) – download link for the dataset

  • horizon (int) – evaluation rollout horizon that should be used with this dataset

Helper function to register dataset link in global dictionary. Also takes a @horizon parameter - this corresponds to the evaluation rollout horizon that should be used during training.

Parameters
  • task (str) – name of task for this dataset

  • dataset_type (str) – type of dataset (usually identifies the dataset source)

  • link (str) – download link for the dataset

  • dataset_size (float) – size of the dataset, in GB