Metadata-Version: 1.0
Name: flopth
Version: 0.0.5
Summary: A program to calculate FLOPs of Pytorch models
Home-page: https://github.com/vra/flopth
Author: Yunfeng Wang
Author-email: wyf.brz@gmail.com
License: MIT Licence
Description: flopth is a program to calculate the FLOPs of Pytorch models, with cli tool and Python API. Support multiple kinds of input, support CPU and GPU, support extra parameters in forward function
Keywords: flopth,Pytorch,Flops,Deep-learning
Platform: any
