Skip to content

Latest commit

 

History

History
8 lines (6 loc) · 440 Bytes

module_tracker.rst

File metadata and controls

8 lines (6 loc) · 440 Bytes

torch.utils.module_tracker

.. automodule:: torch.utils.module_tracker

This utility can be used to track the current position inside an :class:`torch.nn.Module` hierarchy. It 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.

.. autoclass:: torch.utils.module_tracker.ModuleTracker