/pytorch-benchmark

Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption

Primary LanguagePythonApache License 2.0Apache-2.0

Issues