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

[lovegit2021]

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

[PQ-girl]

你好，请问你解决了吗

[lovegit2021]

已经解决了 在 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]

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