inferrence
FengMu1995 opened this issue · 0 comments
FengMu1995 commented
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()