eab5e7909d647e363002b529e9d39911f4a2b028,train2.py,,,#,12

Before Change

    transforms.RandomCrop(train_args["hr_size"]),
    transforms.ToTensor()
])
train_lr_transform = transforms.Compose([
    transforms.ToPILImage(),
    transforms.Scale(train_args["hr_size"] // train_args["scale_factor"], interpolation=3),
    transforms.ToTensor()
])

train_set = datasets.ImageFolder(train_args["train_set_path"], train_ori_transform)
train_loader = DataLoader(train_set, batch_size=64, shuffle=True, num_workers=12,
                          pin_memory=True)

After Change

}

train_set = DatasetFromFolder("data/train", upscale_factor=train_args["scale_factor"],
                              input_transform=transforms.ToTensor(),
                              target_transform=transforms.ToTensor())
train_loader = DataLoader(train_set, batch_size=64, shuffle=True, num_workers=12, pin_memory=True)

Instances