mimicgen package
Contents
mimicgen package#
Subpackages#
- mimicgen.configs package
- mimicgen.datagen package
- mimicgen.env_interfaces package
- mimicgen.envs package
- Subpackages
- mimicgen.envs.robosuite package
- Submodules
- mimicgen.envs.robosuite.coffee module
- mimicgen.envs.robosuite.hammer_cleanup module
- mimicgen.envs.robosuite.kitchen module
- mimicgen.envs.robosuite.mug_cleanup module
- mimicgen.envs.robosuite.nut_assembly module
- mimicgen.envs.robosuite.pick_place module
- mimicgen.envs.robosuite.single_arm_env_mg module
- mimicgen.envs.robosuite.stack module
- mimicgen.envs.robosuite.threading module
- mimicgen.envs.robosuite.three_piece_assembly module
- Module contents
- mimicgen.envs.robosuite package
- Module contents
- Subpackages
- mimicgen.scripts package
- Submodules
- mimicgen.scripts.annotate_subtasks module
- mimicgen.scripts.demo_random_action module
- mimicgen.scripts.download_datasets module
- mimicgen.scripts.generate_config_templates module
- mimicgen.scripts.generate_core_configs module
- mimicgen.scripts.generate_core_training_configs module
- mimicgen.scripts.generate_dataset module
- mimicgen.scripts.generate_robot_transfer_configs module
- mimicgen.scripts.generate_robot_transfer_training_configs module
- mimicgen.scripts.generate_training_configs_for_public_datasets module
- mimicgen.scripts.get_reset_videos module
- mimicgen.scripts.get_source_info module
- mimicgen.scripts.merge_hdf5 module
- mimicgen.scripts.prepare_src_dataset module
- mimicgen.scripts.visualize_subtasks module
- Module contents
- mimicgen.utils package
Module contents#
- mimicgen.register_all_links()#
Record all dataset links in this function.
- mimicgen.register_dataset_link(dataset_type, task, 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
dataset_type (str) – identifies the type of dataset (e.g. source human data, core experiment data, object transfer data)
task (str) – name of task for this dataset
link (str) – download link for the dataset
horizon (int) – evaluation rollout horizon that should be used with this dataset