def __init__(self, wrappee: nn.Module) -> None:
        """Wrap a PyTorch module to make it PackNet compatible.

        :param wrappee: The module to wrap
        """
        super().__init__()
        self.wrappee: nn.Module = PackNetModel.wrap(wrappee)
Wrap a PyTorch module to make it PackNet compatible.

        :param wrappee: The module to wrap
        
avalanche/models/packnet.py