Simple PyTorch profiler that combines DeepSpeed Flops Profiler and TorchInfo
-
Updated
Feb 12, 2023 - Python
#
pytorch-profiler
Here are 3 public repositories matching this topic...
This tutorial was presented at the winter school in Deep Learning at UiT [13/01/2023] (see https://www.nldl.org/winter-school).
-
Updated
Jan 16, 2023 - Shell
Profile your PyTorch model with model-level, layer-level, and operator-level details
-
Updated
Aug 27, 2024 - Jupyter Notebook
Improve this page
Add a description, image, and links to the pytorch-profiler topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the pytorch-profiler topic, visit your repo's landing page and select "manage topics."