class RluDmlabExploreObjectRewardsFew(dmlab_dataset.DMLabDatasetBuilder): """DatasetBuilder for rlu_dmlab_explore_object_rewards_few dataset.""" VERSION = tfds.core.Version('1.0.0') RELEASE_NOTES = { '1.0.0': 'Initial release.', } # pytype: disable=wrong-keyword-args BUILDER_CONFIGS = [ dmlab_dataset.BuilderConfig( name='training_0', task=_TASK, episode_length=_EPISODE_LENGTH), dmlab_dataset.BuilderConfig( name='training_1', task=_TASK, episode_length=_EPISODE_LENGTH), dmlab_dataset.BuilderConfig( name='training_2', task=_TASK, episode_length=_EPISODE_LENGTH) ]
class RluDmlabRoomsSelectNonmatchingObject(dmlab_dataset.DMLabDatasetBuilder): """DatasetBuilder for rlu_dmlab_rooms_select_nonmatching_object dataset.""" VERSION = tfds.core.Version('1.2.0') RELEASE_NOTES = { '1.0.0': 'Initial release.', '1.1.0': 'Added is_last.', '1.2.0': 'BGR -> RGB fix for pixel observations.', } # pytype: disable=wrong-keyword-args BUILDER_CONFIGS = [ dmlab_dataset.BuilderConfig( name='training_0', task=_TASK, episode_length=_EPISODE_LENGTH), dmlab_dataset.BuilderConfig( name='training_1', task=_TASK, episode_length=_EPISODE_LENGTH), dmlab_dataset.BuilderConfig( name='training_2', task=_TASK, episode_length=_EPISODE_LENGTH) ]
class RluDmlabSeekavoidArena01(dmlab_dataset.DMLabDatasetBuilder): """DatasetBuilder for rlu_dmlab_seekavoid_arena01 dataset.""" VERSION = tfds.core.Version('1.1.0') RELEASE_NOTES = { '1.0.0': 'Initial release.', '1.1.0': 'Added is_last.', } # pytype: disable=wrong-keyword-args BUILDER_CONFIGS = [ dmlab_dataset.BuilderConfig(name=name, task=_TASK, episode_length=_EPISODE_LENGTH) for name in _CONFIG_NAMES ]
class RluDmlabSeekavoidArena01(dmlab_dataset.DMLabDatasetBuilder): """DatasetBuilder for rlu_dmlab_seekavoid_arena01 dataset.""" VERSION = tfds.core.Version('1.0.0') RELEASE_NOTES = { '1.0.0': 'Initial release.', } _TASK = 'seekavoid_arena_01' _EPISODE_LENGTH = 301 # pytype: disable=wrong-keyword-args BUILDER_CONFIGS = [ dmlab_dataset.BuilderConfig(name='training_0', task=_TASK, episode_length=_EPISODE_LENGTH), # TODO(b/188125902): Enable the rest of the configs once we double check # that the first one works. # dmlab_dataset.TaskConfig(name='training_1', # task=_TASK, # episode_length=_EPISODE_LENGTH), # dmlab_dataset.TaskConfig(name='training_2', # task=_TASK, # episode_length=_EPISODE_LENGTH), # dmlab_dataset.TaskConfig(name='snapshot_0_eps_0.0', # task=_TASK, # episode_length=_EPISODE_LENGTH), # dmlab_dataset.TaskConfig(name='snapshot_1_eps_0.0', # task=_TASK, # episode_length=_EPISODE_LENGTH), # dmlab_dataset.TaskConfig(name='snapshot_0_eps_0.01', # task=_TASK, # episode_length=_EPISODE_LENGTH), # dmlab_dataset.TaskConfig(name='snapshot_1_eps_0.01', # task=_TASK, # episode_length=_EPISODE_LENGTH), # dmlab_dataset.TaskConfig(name='snapshot_0_eps_0.25', # task=_TASK, # episode_length=_EPISODE_LENGTH), # dmlab_dataset.TaskConfig(name='snapshot_1_eps_0.25', # task=_TASK, # episode_length=_EPISODE_LENGTH), ]