/pytorch-estimate-flops

Estimate/count FLOPS for a given neural network using pytorch

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

Watchers