Make models eval mode during test time
(self)
| 88 | self.print_networks(opt.verbose) |
| 89 | |
| 90 | def eval(self): |
| 91 | """Make models eval mode during test time""" |
| 92 | for name in self.model_names: |
| 93 | if isinstance(name, str): |
| 94 | net = getattr(self, 'net' + name) |
| 95 | net.eval() |
| 96 | |
| 97 | def test(self): |
| 98 | """Forward function used in test time. |