3abe8e680e1f8000dbc6bf2c7aa11bd53e513368,train_model.py,,main,#,17

Before Change

        npatches = max(0, (opts.n_iter - model.count_iter)*opts.batch_size),
        verbose = True,
    )
    dataloader_train = torch.utils.data.DataLoader(
        ds_patch,
        batch_size = opts.batch_size,
    )

    dataloader_test = None
    if opts.checkpoint_testing:
        raise NotImplementedError

After Change

        df_losses = pd.read_csv(path_losses_csv)
        logger.info("History loaded from: {:s}".format(path_losses_csv))

    opts.__dict__["npatches"] = max(0, (opts.n_iter - model.count_iter)*opts.batch_size)
    dataloader_train = get_dataloader(opts)
    dataloader_test = None
    

Instances