search

Eric-mingjie / rethinking-network-pruning

Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)
MIT License
1.51k stars 293 forks source link

count_flops #41

Open kaijieshi7 opened 4 years ago

kaijieshi7 commented 4 years ago

count_flops：你的flops计算的代码有问题。

Eric-mingjie commented 4 years ago

哪里有问题

DachaoLin commented 4 years ago

Why BN layer flops is just 2 not 4？ BN layer : \gamma*(x-\mu)/\sigma+\beta