神经网络模型中-model/model.name_module()的区别
标题 model / model.name_module()的区别:对于已经训练好的网络模型,为了方便后续使用和提取某一层的参数,在储存模型时,我们会对网络每一层进行命名+储存参数数据。所以我们可以对每一层进行操作,例如剪枝操作。在判断当前层时,使用:for name, m0 in model.name_module(): if isinstance(m0, Conv): ....(argument you need) 那么到底print(model)和print(model.name_m
原创
2020-05-19 22:18:20 ·
3722 阅读 ·
0 评论