code: def __init__(
      self,
      model_variable,
      total_classes,
      embed_dimension,
      datatype=torch.float,
      deviceType="cuda",
      pinType=False,
  ):
      self.model_variable = model_variable
      self.total_classes = total_classes
      self.embed_dimension = embed_dimension
      self.dtype = datatype
      self.deviceType = deviceType
      self.pinType = pinType
  
imports: import torch
  import os
  import numpy
label_desc: Use image node_id 1 to create a new PytorchModelFactory object from inherited base classes: _PytorchCommon with example: obj = PytorchModelFactory(model_variable, total_classes, embed_dimension, datatype, deviceType, pinType)