predict.py 文件源码

def get_loader(df, transformations):
    dset_val = KaggleAmazonJPGDataset(df, paths.train_jpg, transformations, divide=False)

    loader_val = DataLoader(dset_val,
                            batch_size=batch_size,
                            num_workers=12,
                            pin_memory=True,)
    return loader_val