mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-25 16:14:55 +08:00
458cc7213b
Fixes #155026 Pull Request resolved: https://github.com/pytorch/pytorch/pull/155702 Approved by: https://github.com/sekyondaMeta, https://github.com/svekars Co-authored-by: Svetlana Karslioglu <svekars@meta.com>
12 lines
442 B
Markdown
12 lines
442 B
Markdown
# torch.utils.module_tracker
|
|
```{eval-rst}
|
|
.. 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.
|
|
|
|
```{eval-rst}
|
|
.. autoclass:: torch.utils.module_tracker.ModuleTracker
|
|
```
|