code: def state_dict(self):
      if "state_dict" in amp.__dict__:
          return amp.state_dict()
  
file_path: pytorch-image-models/timm/utils/cuda.py
imports: import torch
  from .clip_grad import dispatch_clip_grad
label_desc: Use image node_id 2 for calling the ApexScaler obj's underlying member method code with example usage: obj.state_dict() and returns: amp