Make models eval mode
(self)
| 146 | net.train() |
| 147 | |
| 148 | def eval(self): |
| 149 | """Make models eval mode""" |
| 150 | for name in self.model_names: |
| 151 | if isinstance(name, str): |
| 152 | net = getattr(self, name) |
| 153 | net.eval() |
| 154 | |
| 155 | def test(self): |
| 156 | """Forward function used in test time. |
no outgoing calls