def create_dataset_train_ssl(model):
    """Helper method to apply the correct transform for ssl.

    Args:
        model:
            Model class for which to select the transform.
    """
    model_to_transform = {
        BarlowTwinsModel: byol_transform,
        BYOLModel: byol_transform,
        DCL: simclr_transform,
        DCLW: simclr_transform,
        DINOModel: dino_transform,
        FastSiamModel: fast_siam_transform,
        MAEModel: mae_transform,
        MSNModel: msn_transform,
        MocoModel: simclr_transform,
        NNCLRModel: simclr_transform,
        PMSNModel: msn_transform,
        SimCLRModel: simclr_transform,
        SimMIMModel: simmim_transform,
        SimSiamModel: simsiam_transform,
        SwaVModel: swav_transform,
        SwaVQueueModel: swav_transform,
        SMoGModel: smog_transform,
        TiCoModel: byol_transform,
        VICRegModel: vicreg_transform,
        VICRegLModel: vicregl_transform,
    }
    transform = model_to_transform[model]
    return LightlyDataset(input_dir=path_to_train, transform=transform)
Helper method to apply the correct transform for ssl.

    Args:
        model:
            Model class for which to select the transform.
    
docs/source/getting_started/benchmarks/imagenette_benchmark.py