【Pytorch参数初始化】pytorch模型参数默认初始化init问题

按照书上操作来初始化模型参数时遇到了问题

在这里插入图片描述
书上写的挺好,但是实际中就会出问题:
object is not subscriptable在这里插入图片描述

解决方案其实很简单

出错的根本原因是,net这个对象没有可以用下标表示的元素
我们首先print一下这个net有啥:
在这里插入图片描述
这是一个线性的神经网络,两个输入一个输出
所以我们只要把出错的那一行改成:

init.normal_(net.linear.weight, mean=0, std=0.01)

就ok啦

猜你喜欢

转载自blog.csdn.net/QinZheng7575/article/details/106753988