class_docstr: Track optimizer progress.
  
  Args:
      step: Tracks ``optimizer.step`` calls.
      zero_grad: Tracks ``optimizer.zero_grad`` calls.
class_docstr_tok: ["Track","optimizer","progress",".","Args",":","step",":","Tracks","``","optimizer.step","``","calls",".","zero_grad",":","Tracks","``","optimizer.zero_grad","``","calls","."]
code: def load_state_dict(self, state_dict: dict) -> None:
      self.step.load_state_dict(state_dict["step"])
      self.zero_grad.load_state_dict(state_dict["zero_grad"])
  
imports: from dataclasses import asdict, dataclass, field
  from typing import Type
  from typing_extensions import override
label_desc: Use image node_id 4 for calling the _OptimizerProgress obj's underlying member method code with example usage: obj.load_state_dict(state_dict) without return types