code: def __init__(
      self, input_dimension, output_dimension, device, **kwargs
  ):
      self._input_dimension = input_dimension
      self._output_dimension = output_dimension
      self._model = None
      self._kwargs = kwargs
      super(BaseAutoModel, self).__init__(device)
  
imports: import copy
  import logging
  import typing
  import torch
  import torch.nn.functional
  from copy import deepcopy
  from autogl.module.hpo.auto_module import AutoModule
label_desc: Use image node_id 1 to create a new BaseAutoModel object from inherited base classes: AutoModule with example: obj = BaseAutoModel(input_dimension, output_dimension, device)