eab5e7909d647e363002b529e9d39911f4a2b028,train2.py,,train,#,36

Before Change

        train_bar = tqdm(train_loader)
        for data in train_bar:
            hr_imgs, _ = data
            lr_imgs = Variable(torch.stack([train_lr_transform(img) for img in hr_imgs], 0)).cuda()
            hr_imgs = Variable(hr_imgs).cuda()
            gen_hr_imgs = g(lr_imgs)

            // update d

After Change

        train_bar = tqdm(train_loader)
        for hr_imgs, lr_imgs in train_bar:
            hr_imgs = Variable(hr_imgs).cuda()
            lr_imgs = Variable(lr_imgs).cuda()
            gen_hr_imgs = g(lr_imgs)

            // update d
            d.zero_grad()

Instances