Estimate/count FLOPS for a given neural network using pytorch - View it on GitHub