(self, mode: bool = True, recursive: bool = True)
| 54 | self.enabled = False |
| 55 | |
| 56 | def train(self, mode: bool = True, recursive: bool = True) -> None: |
| 57 | super().train(mode, recursive) |
| 58 | if mode: |
| 59 | self.enable() |
| 60 | else: |
| 61 | self.disable() |
| 62 | |
| 63 | @abstractmethod |
| 64 | def forward(self, x): |