code: def __setstate__(self, state):
      super(DeepSpeedCPUAdam, self).__setstate__(state)
      for group in self.param_groups:
          group.setdefault("amsgrad", False)
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/deepspeed/ops/adam/cpu_adam.py
imports: import torch
  from cpuinfo import get_cpu_info
  from ..op_builder import CPUAdamBuilder
  from deepspeed.utils import logger
  from deepspeed.utils.logging import should_log_le
label_desc: Use image node_id 3 for calling the DeepSpeedCPUAdam obj's underlying member method code with example usage: obj.__setstate__(state) without return types