Skip to content

task

task

Configurations for the experiment task. The task defines what data the experiment will use at runtime, but not what data is in the dataset (see DatasetSchema for that). This design is motivated by the fact that a single dataset can be used for multiple tasks, each requiring different data.

ActionSpaceConfig dataclass

ActionSpaceConfig(_target_='versatil.data.task.ActionSpace', actions_metadata=dict(), use_gripper_class_weights=False, denoise_actions=True, denoising_percentile=15.0)

Configuration for action space.

Attributes:

Name Type Description
_target_ str

Import path instantiated by Hydra.

actions_metadata dict[str, Any]

Dict of all action metadata, indexed by zarr store key. Values are OnTheFlyActionMetadataConfig or PrecomputedActionMetadataConfig subclasses.

use_gripper_class_weights bool

Whether to use class weights for binary gripper.

denoise_actions bool

Whether to apply denoising to actions.

denoising_percentile float

Percentile for denoising threshold.

ObservationSpaceConfig dataclass

ObservationSpaceConfig(_target_='versatil.data.task.ObservationSpace', observations_metadata=dict())

Configuration for observation space.

Attributes:

Name Type Description
_target_ str

Import path instantiated by Hydra.

observations_metadata dict[str, Any]

Dict of all observation metadata, indexed by zarr store key. Values are ObservationMetadataConfig subclasses, RGBCameraMetadataConfig, or DepthCameraMetadataConfig.

TaskSpaceConfig dataclass

TaskSpaceConfig(_target_='versatil.data.task.TaskSpace', dataset_schema=MISSING, dataloader=MISSING, action_space=MISSING, observation_space=MISSING, observation_horizon=1, prediction_horizon=16)

Task space specific configuration for the experiment run.

Attributes:

Name Type Description
_target_ str

Import path instantiated by Hydra.

dataset_schema Any

Dataset schema configuration, defining what dataset and zarr store the task uses.

dataloader DataLoaderConfig

Data loading and preprocessing configuration.

action_space ActionSpaceConfig

Action space configuration used by the task at runtime.

observation_space ObservationSpaceConfig

Observation space configuration used by the task at runtime.

observation_horizon int

Number of history timesteps to include.

prediction_horizon int

Number of timesteps to predict (action chunk size).