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
Atlas, ML Training Observatory that monitors PyTorch training runs in real-time, automatically detects 30+ training issues, and diagnoses problem with an adaptive AI agent powered by Crusoe. (Winner for Best use of Crusoe)
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."