• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1torch.utils.module_tracker
2===================================
3.. automodule:: torch.utils.module_tracker
4
5This utility can be used to track the current position inside an :class:`torch.nn.Module` hierarchy.
6It can be used within other tracking tools to be able to easily associate measured quantities to user-friendly names. This is used in particular in the FlopCounterMode today.
7
8.. autoclass:: torch.utils.module_tracker.ModuleTracker
9