GitHub repository: 1adrianb/pytorch-estimate-flops
Primary language: Python
Description: Estimate/count FLOPS for a given neural network using pytorch