Initializes the model.
(self, *args, model_mode: str = MODEL_MODE_TRAIN, **kwargs)
| 58 | # pylint: enable=attribute-defined-outside-init |
| 59 | |
| 60 | def init(self, *args, model_mode: str = MODEL_MODE_TRAIN, **kwargs): |
| 61 | """Initializes the model.""" |
| 62 | module = self.clone(model_mode=model_mode) |
| 63 | kwargs["model_mode"] = model_mode |
| 64 | return nn.Module.init(module, *args, **kwargs) |
| 65 | |
| 66 | def apply(self, *args, model_mode: str = MODEL_MODE_TRAIN, **kwargs): |
| 67 | """Applies the model.""" |