search

Eric-mingjie / network-slimming

Network Slimming (Pytorch) (ICCV 2017)
MIT License
907 stars 214 forks source link

If the remaining channel for a layer is zero, it reports zero division error #80

Open Neil-untitled opened 2 years ago

Neil-untitled commented 2 years ago

I trained vgg with cifar100 dataset and there are a few layers with zero remaining channels. As a result I got a zerodivisionerror in vgg.py part when initializing weights. Can someone help me plz?

Eric-mingjie commented 2 years ago

i think you are using a large prune ratio, try smaller prune ratio e.g. 0.1, 0.2 to see if it still exists?

Neil-untitled commented 2 years ago

Hi Mingjie, Many thanks for your reply. I will try with smaller ratio again later. Informative suggestion.

Regards Haoran Geng

---Original--- From: @.> Date: Wed, Dec 15, 2021 07:33 AM To: @.>; Cc: @.**@.>; Subject: Re: [Eric-mingjie/network-slimming] If the remaining channel for a layer is zero, it reports zero division error (Issue #80)

i think you are using a large prune ratio, try smaller prune ratio e.g. 0.1, 0.2 to see if it still exists?

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe. Triage notifications on the go with GitHub Mobile for iOS or Android.