search

Eric-mingjie / network-slimming

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

剪枝后保存的权重文件和newmodel加载不上 #78

Open lovegit2021 opened 2 years ago

lovegit2021 commented 2 years ago

您好，我剪枝resnet164后保存为pruned.pth.tar和新的cfg，但是只用newmodel加载的时候无法加载pruned.pth.tar，报错为参数尺寸不匹配，想请教您这是因为channel_selection没有处理好吗?

PQ-girl commented 2 years ago

你好，请问你解决了吗

lovegit2021 commented 2 years ago

已经解决了

在 2022-03-10 11:50:33，"PQ-girl" @.***> 写道：

你好，请问你解决了吗

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

YHLymj commented 2 years ago

您好！在进行剪枝时，传入的模型是稀疏训练保存的模型还是baseline训练的模型？ python vggprune.py --dataset cifar10 --depth 19 --percent 0.7 --model [PATH TO THE MODEL]