Open FengMu1995 opened 3 years ago
why exits "self.netG.train()" in SRGAN_model.SRGANModel.test函数中as follows:，推理应该只包括test部分吧?
def test(self): self.netG.eval() with torch.no_grad(): self.fake_H = self.netG(self.var_L) self.netG.train()
why exits "self.netG.train()" in SRGAN_model.SRGANModel.test函数中as follows:，推理应该只包括test部分吧?
def test(self): self.netG.eval() with torch.no_grad(): self.fake_H = self.netG(self.var_L) self.netG.train()