from .train_barlowtwins import pretrain_barlowtwins, BarlowTwins
from .train_byol import pretrain_byol, BYOL
from .train_dino import pretrain_dino, DINO
from .train_simclr import pretrain_simclr, SimCLR
from .train_supervised import pretrain_supervised, Supervised
from .train_swav import pretrain_swav, SwaV
from .train_vicreg import pretrain_vicreg, VICReg