code: def __init__(self, module):
      super(SingleGPU, self).__init__()
      self.module = module
  
file_path: insightface/parsing/dml_csr/utils/utils.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  import torchvision
  import torch
  from PIL import Image
  from torch import nn
label_desc: Use image node_id 1 to create a new SingleGPU object from inherited base classes: nn with example: obj = SingleGPU(module)