class_docstr: Base space class of NAS module. Defining space containing all models.
  Please use mutables to define your whole space. Refer to
  `https://nni.readthedocs.io/en/stable/NAS/WriteSearchSpace.html`
  for detailed information.
  
  Parameters
  ----------
  init: bool
      Whether to initialize the whole space. Default: `False`
docstr: Return a BoxModel which wrap self as a model
  Used to pass to trainer
  To use this function, must contain `input_dim` and `output_dim`
docstr_tok: ["Return","a","BoxModel","which","wrap","self","as","a","model","Used","to","pass","to","trainer","To","use","this","function",",","must","contain","`","input_dim","`","and","`","output_dim","`"]
code: def wrap(self):
      """
      Return a BoxModel which wrap self as a model
      Used to pass to trainer
      To use this function, must contain `input_dim` and `output_dim`
      """
      device = next(self.parameters()).device
      return BoxModel(self, device)
  
imports: from abc import abstractmethod
  import torch.nn
  from nni.nas.pytorch import mutables
  from nni.nas.pytorch.fixed import FixedArchitecture
  import json
  from copy import deepcopy
  import torch
  from ...model import BaseAutoModel
  from ....utils import get_logger
  from ..utils import get_hardware_aware_metric
label_desc: Use image node_id 8 for calling the BaseSpace obj's underlying member method code with example usage: obj.wrap() and returns: BoxModel