Skip to content

contrastive

contrastive

ContrastiveTrainer(spec: ExecutionSpec, base: Optional[Node] = None)

Bases: BaseTrainer[BaseTrainerConfig, M], Generic[M]

Contrastive trainer scaffold.

BackbonedContrastiveTrainer(spec: ExecutionSpec, base: Optional[Node] = None)

Bases: BackbonedTrainer, ContrastiveTrainer[Module]

Contrastive trainer scaffold for backboned training.