c24a418b1e40dab347dcdcacb4f0755f905e9f79,codes/models/SRGAN_model.py,SRGANModel,test,#SRGANModel#,179

Before Change

    def test(self):
        self.netG.eval()
        for k, v in self.netG.named_parameters():
            v.requires_grad = False
        self.fake_H = self.netG(self.var_L)
        for k, v in self.netG.named_parameters():
            v.requires_grad = True
        self.netG.train()

    def get_current_log(self):
        return self.log_dict

After Change

    def test(self):
        self.netG.eval()
        with torch.no_grad():
            self.fake_H = self.netG(self.var_L)
        self.netG.train()

    def get_current_log(self):
        return self.log_dict

Instances