pytorch-benchmark
pytorch-benchmark copied to clipboard
Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption
Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption