LukasHedegaard/pytorch-benchmark
Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption
PythonApache-2.0
Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption
PythonApache-2.0