Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption
What is the LukasHedegaard/pytorch-benchmark GitHub project? Description: "Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption ". Written in Python. Explain what it does, its main use cases, key features, and who would benefit from using it.
Question is copied to clipboard — paste it after the AI opens.
Clone via HTTPS
Clone via SSH
Download ZIP
Download master.zipReport bugs or request features on the pytorch-benchmark issue tracker:
Open GitHub Issues