Unitary CLIP Pipeline

Pipeline for contrastive pre-training of an unitary encoder

source

UnitaryCLIPPipeline

 UnitaryCLIPPipeline (model:torch.nn.modules.module.Module,
                      device:torch.device)

A PipelineIO class providing basic pytorch model training functionality.

Back to top