code: def __init__(self, net):
      super().__init__()
      self.net = net
      self.cpu_state = None
      self.cuda_in_fwd = None
      self.gpu_devices = None
      self.gpu_states = None
  
imports: import torch
  import torch.nn
  from operator import itemgetter
  from torch.autograd.function import Function
  from torch.utils.checkpoint import get_device_states, set_device_states
label_desc: Use image node_id 1 to create a new Deterministic object from inherited base classes: nn with example: obj = Deterministic(net)