ModalitySpec#

class torch_brain.registry.ModalitySpec(*args, **kwargs)[source]#

Bases: object

Specification for a modality.

dim#: Dimension for this modality

type#: DataType enum specifying the data type

loss_fn#: Name of loss function to use for this modality

timestamp_key#: Key to access timestamps in the data object

value_key#: Key to access values in the data object

id#: Unique numeric ID assigned to this modality