90172e414c9bc8c03a537065f3c8d8133c9dae26,train2.py,,,#,17

Before Change

set5 = datasets.ImageFolder(train_args["set5_path"], val_ori_transform)
set14 = datasets.ImageFolder(train_args["set14_path"], val_ori_transform)
bsd100_loader = DataLoader(bsd100, batch_size=1, num_workers=12, pin_memory=True)
set5_loader = DataLoader(set5, batch_size=1, num_workers=12, pin_memory=True)
set14_loader = DataLoader(set14, batch_size=1, num_workers=12, pin_memory=True)
val_loader = {"bsd100": bsd100_loader, "set5": set5_loader, "set14": set14_loader}

After Change

from loss import GeneratorLoss
from model import Generator, Discriminator

train_args = {
    "train_batch_size": 64,
    "scale_factor": 4,  // should be power of 2
    "ckpt_path": "epochs",
}


val_lr_transform = LRTransformTest(train_args["scale_factor"])
val_display_transform = transforms.Compose([

Instances