mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
8 lines
440 B
ReStructuredText
8 lines
440 B
ReStructuredText
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
|