Sets the model to evaluation mode.
(self)
| 131 | self._model.train() |
| 132 | |
| 133 | def eval(self): |
| 134 | """Sets the model to evaluation mode.""" |
| 135 | self.training = False |
| 136 | self._model.eval() |
| 137 | |
| 138 | def backward(self, loss: torch.Tensor): |
| 139 | """ |
no outgoing calls
no test coverage detected