Make models eval mode during test time
(self)
| 110 | pass |
| 111 | |
| 112 | def eval(self): |
| 113 | """Make models eval mode during test time""" |
| 114 | for name in self.model_names: |
| 115 | if isinstance(name, str): |
| 116 | net = getattr(self, 'net' + name) |
| 117 | net.eval() |
| 118 | |
| 119 | def test(self): |
| 120 | """Forward function used in test time. |