training_entry_point.py

def main():
    params = create_default_parser(__file__).parse_args()

    super_seed(params.seed)

    loaders, meta = create_data_loaders(dataset_name=params.dataset,
                                        dataset_root=params.dataset_path,
                                        num_workers=params.num_workers,
                                        batch_size=params.batch_size,
                                        sample_transformer=transforms.Compose([
                                            transforms.ToTensor(),
                                            ToThreeChannels(),
                                            transforms.Normalize([0.5]*3, [0.5]*3)
                                        ]))

    params.n_classes = len(meta['classes'])

    trainer = pl.Trainer(**make_trainer_parameters(params))

    trainer.fit(model=BasicCNNExperiment(params),
                train_dataloader=loaders['train'],
                val_dataloaders=loaders['valid'])