mmlearn.tasks.contrastive_pretraining.ModuleKeySpec¶

class ModuleKeySpec(encoder_key=None, head_key=None, postprocessor_key=None)[source]¶

Bases: object

Module key specification for mapping modules to modalities.

Methods

Attributes

encoder_key: Optional[str] = None¶: The key of the encoder module. If not provided, the modality name is used.

head_key: Optional[str] = None¶: The key of the head module. If not provided, the modality name is used.

postprocessor_key: Optional[str] = None¶: The key of the postprocessor module. If not provided, the modality name is used.