class_docstr: A base class to support exporting through ONNX
docstr: Override this in DistributedDataParallel models
code: def cpu_model(self) -> "ModelBase":
      """
      Override this in DistributedDataParallel models
      """
      # This is not ideal but makes exporting simple
      return deepcopy(self).cpu()
  
file_path: ReAgent/reagent/models/base.py
imports: from copy import deepcopy
  from typing import Any, Optional
  import torch.nn
  from reagent.core import types
label_desc: Use image node_id 5 for calling the ModelBase obj's underlying member method code with example usage: obj.cpu_model() and returns: deepcopy