split-layer

A tool to measure forward and backward time for each layer in PyTorch

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install split-layer

Releases

Version Released Bullseye
Python 3.9
Bookworm
Python 3.11
Files
0.1.1 2021-05-05  
0.1.0 2021-05-05  
0.0.10 2021-03-31  

Issues with this package?

Page last updated 2025-06-27 16:23:47 UTC