/pytorch-network-slimming

network slimming for pytorch models, support user defined models with Linear/Conv(groups=1)/BN

Primary LanguagePythonMIT LicenseMIT

Watchers