Tencent/Real-SR

inferrence

FengMu1995 opened this issue · 0 comments

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