神经网络模型中-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 ·
3528 阅读 ·
0 评论