def METHOD_NAME(self, mode: bool = True) -> "Model":
    """Deactivate inference mode."""
    super().METHOD_NAME(mode)
    if mode:
        for task in self._tasks:
            task.train_eval()
    return self