class_docstr: Optim wrapper that implements rate.
class_docstr_tok: ["Optim","wrapper","that","implements","rate","."]
docstr: Return state_dict.
docstr_tok: ["Return","state_dict","."]
code: def state_dict(self):
      """Return state_dict."""
      return {
          "_step": self._step,
          "warmup": self.warmup,
          "factor": self.factor,
          "model_size": self.model_size,
          "_rate": self._rate,
          "optimizer": self.optimizer.state_dict(),
      }
  
file_path: espnet/espnet/nets/pytorch_backend/transformer/optimizer.py
imports: import torch
label_desc: Use image node_id 6 for calling the NoamOpt obj's underlying member method code with example usage: obj.state_dict() and returns: dict