pytorchures

Python profiler for pytorch models. Wraps each layer and measures it's execution time.

MIT License

Downloads
197
Stars
0
View Code on GitHub
Ecosystems: PyTorch, Python
Package Rankings
Top 34.07% on Pypi.org
Related Projects
pytorch-lightning-snippets

A collection of code snippets for my PyTorch Lightning projects

22 Aug 2020 107
causal-tracer

Causal tracing for language models

16 Aug 2023 8
ai-toolkit

Unifying all code detailing learned ML concepts for easy reuse.

16 Dec 2018 14
keras2tensorflow

Tutorial on running keras model in C++ and python tensorflow

30 Oct 2018 11
torchprep

21 Jul 2022 9
Complex-YOLOv4-Pytorch

The PyTorch Implementation based on YOLOv4 of the paper: "Complex-YOLO: Real-time 3D Object Detec...

03 Jul 2020 1,234
torchextractor

Feature extraction made simple with torchextractor

03 Mar 2021 99
torchview

torchview: visualize pytorch models

08 Nov 2022 803
torchstat

Model analyzer in PyTorch

26 Oct 2018 1,460
line_profiler

Line-by-line profiling for Python

10 Dec 2019 2,678
pyinstrument

🚴 Call stack profiler for Python. Shows you why your code is slow!

13 Mar 2014 6,094
torchnnprofiler

Context Manager to profile the forward and backward times of PyTorch's nn.Module

22 Oct 2022 82
torchinfo

View model summaries in PyTorch!

16 Mar 2020 2,523
few-shot-open-set

Implementation of Open-Set Likelihood Maximization for Few-Shot Learning

18 Jun 2022 36
thinc

🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

16 Oct 2014 2,817