pytorch-benchmark
0.3.6
43.41k
Easily benchmark PyTorch model FLOPs, latency, throughput, max allocated memory and energy consumption in one go.
Easily benchmark PyTorch model FLOPs, latency, throughput, max allocated memory and energy consumption in one go.