Skip to content

plugins.habitat_plugin.habitat_utils#

[view_source]

construct_env_configs#

construct_env_configs(config: Config, allow_scene_repeat: bool = False) -> List[Config]

[view_source]

Create list of Habitat Configs for training on multiple processes To allow better performance, dataset are split into small ones for each individual env, grouped by scenes.

Parameters

  • config : configs that contain num_processes as well as information necessary to create individual environments.
  • allow_scene_repeat: if True and the number of distinct scenes in the dataset is less than the total number of processes this will result in scenes being repeated across processes. If False, then if the total number of processes is greater than the number of scenes, this will result in a RuntimeError exception being raised.

Returns

List of Configs, one for each process.