code: def __init__(self, model: LightningModule, layer: str):
      super().__init__()
  
      self.model = model
      self.layer = layer
  
      self._module, self._hook = self._make_hook()
      self._handle = None
      self._out = None
  
imports: from typing import Any, Optional
  from pytorch_lightning import LightningModule
  from torch.utils.data import DataLoader, Sampler
  import flash
  from flash.core.data.io.input import InputBase
  from flash.core.data.io.input_transform import InputTransform
  from flash.core.model import Task
label_desc: Use image node_id 1 to create a new Embedder object from inherited base classes: Task with example: obj = Embedder(model, layer)