code: def __init__(self, model: nn.Module):
      super().__init__()
      self.model = model
  
file_path: icevision/icevision/models/mmseg/fastai/callbacks.py
imports: from icevision.imports import *
  from icevision.utils import *
  from icevision.core import *
  from icevision.data import *
  from icevision.engines.fastai import *
  from icevision.models.mmseg.utils import *
  from icevision.models.mmseg.common.prediction import convert_raw_predictions
label_desc: Use image node_id 1 to create a new _ModelWrap object from inherited base classes: nn with example: obj = _ModelWrap(model)