Simple PyTorch profiler that combines DeepSpeed Flops Profiler and TorchInfo
-
Updated
Feb 12, 2023 - Python
Simple PyTorch profiler that combines DeepSpeed Flops Profiler and TorchInfo
Add a description, image, and links to the pytorch-profiler topic page so that developers can more easily learn about it.
To associate your repository with the pytorch-profiler topic, visit your repo's landing page and select "manage topics."