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
Assignments for Memory and Storage Systems (Graduate Course, NYCU)
This tutorial was presented at the winter school in Deep Learning at UiT [13/01/2023] (see https://www.nldl.org/winter-school).
Profile your PyTorch model with model-level, layer-level, and operator-level details
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."