robomimic package
Contents
robomimic package#
Subpackages#
- robomimic.algo package
- robomimic.config package
- Submodules
- robomimic.config.base_config module
- robomimic.config.bc_config module
- robomimic.config.bcq_config module
- robomimic.config.config module
- robomimic.config.cql_config module
- robomimic.config.gl_config module
- robomimic.config.hbc_config module
- robomimic.config.iql_config module
- robomimic.config.iris_config module
- robomimic.config.td3_bc_config module
- Module contents
- robomimic.envs package
- robomimic.models package
- Submodules
- robomimic.models.base_nets module
- robomimic.models.distributions module
- robomimic.models.obs_core module
- robomimic.models.obs_nets module
- robomimic.models.policy_nets module
- robomimic.models.transformers module
- robomimic.models.vae_nets module
- robomimic.models.value_nets module
- Module contents
- robomimic.utils package
- Submodules
- robomimic.utils.dataset module
- robomimic.utils.env_utils module
- robomimic.utils.file_utils module
- robomimic.utils.hyperparam_utils module
- robomimic.utils.log_utils module
- robomimic.utils.loss_utils module
- robomimic.utils.obs_utils module
- robomimic.utils.python_utils module
- robomimic.utils.tensor_utils module
- robomimic.utils.test_utils module
- robomimic.utils.torch_utils module
- robomimic.utils.train_utils module
- robomimic.utils.vis_utils module
- Module contents
Submodules#
robomimic.macros module#
Set of global variables shared across robomimic
robomimic.macros_private module#
Set of global variables shared across robomimic
Module contents#
- robomimic.register_all_links()#
Record all dataset links in this function.
- robomimic.register_all_momart_links()#
Record all dataset links in this function.
- robomimic.register_dataset_link(task, dataset_type, hdf5_type, link, horizon)#
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
- robomimic.register_momart_dataset_link(task, dataset_type, link, dataset_size)#
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