search

Eric-mingjie / network-slimming

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

关于FLOPs计算 #60

Closed arthur0219 closed 3 years ago

arthur0219 commented 3 years ago

作者您好，想问一下对剪枝后的模型，怎么查看参数量和FLOPs呢？在剪枝后模型上用普通的指令会报错

Eric-mingjie commented 3 years ago

https://github.com/Eric-mingjie/rethinking-network-pruning/blob/master/imagenet/l1-norm-pruning/compute_flops.py

arthur0219 commented 3 years ago

Ok, thanks a lot