Sets the node in training mode. :param mode: Turn training on or off. :return: ``self`` as specified in ``torch.nn.Module``.
(self, mode: bool = True)
| 479 | self.monitors[monitor].reset_state_variables() |
| 480 | |
| 481 | def train(self, mode: bool = True) -> "torch.nn.Module": |
| 482 | # language=rst |
| 483 | """ |
| 484 | Sets the node in training mode. |
| 485 | |
| 486 | :param mode: Turn training on or off. |
| 487 | |
| 488 | :return: ``self`` as specified in ``torch.nn.Module``. |
| 489 | """ |
| 490 | self.learning = mode |
| 491 | return super().train(mode) |
no outgoing calls
no test coverage detected