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()