(self)
| 119 | self.model_ema(decay=self.ema_decay) |
| 120 | |
| 121 | def test(self): |
| 122 | if hasattr(self, 'net_g_ema'): |
| 123 | self.net_g_ema.eval() |
| 124 | with torch.no_grad(): |
| 125 | self.output = self.net_g_ema(self.lq) |
| 126 | else: |
| 127 | self.net_g.eval() |
| 128 | with torch.no_grad(): |
| 129 | self.output = self.net_g(self.lq) |
| 130 | self.net_g.train() |
| 131 | |
| 132 | def test_selfensemble(self): |
| 133 | # TODO: to be tested |
no outgoing calls
no test coverage detected