class_docstr: Class to track progress during the loop. Includes state_dict/load_state_dict for convenience for checkpointing.
class_docstr_tok: ["Class","to","track","progress","during","the","loop",".","Includes","state_dict\/load_state_dict","for","convenience","for","checkpointing","."]
docstr: Returns a state_dict of a Progress instance in accordance with Stateful protocol.
docstr_tok: ["Returns","a","state_dict","of","a","Progress","instance","in","accordance","with","Stateful","protocol","."]
code: def state_dict(self) -> Dict[str, Any]:
      """Returns a state_dict of a Progress instance in accordance with Stateful protocol."""
      return {
          "num_epochs_completed": self._num_epochs_completed,
          "num_steps_completed": self._num_steps_completed,
          "num_steps_completed_in_epoch": self._num_steps_completed_in_epoch,
      }
  
file_path: tnt/torchtnt/utils/progress.py
imports: from collections.abc import Sized
  from typing import Any, Dict, Iterable, Optional
label_desc: Use image node_id 7 for calling the Progress obj's underlying member method code with example usage: obj.state_dict() and returns: dict