predict.py 文件源码

def get_test_loader(test_images, transformations):
    dset_test = KaggleAmazonTestDataset(test_images, paths.test_jpg, '.jpg', transformations, divide=False)
    loader_val = DataLoader(dset_test,
                            batch_size=batch_size,
                            num_workers=12,
                            pin_memory=True)
    return loader_val