nn.Parameter生成含Nan(日常记录)

训练模型的时候发现某一步的输出结果一下子一片都变成Nan了,经过排查发现是可训练参数生成时就有Nan,真是无语 = =。

nn.Parameter生成含Nan(日常记录)_第1张图片问题源头在这里:

self.u = nn.Parameter(torch.Tensor(x, y))

翻遍了全网,在一个老哥那里发现了同样的问题,文章链接如下

关于Pytorch中Parameter的nan | Weekly Review

看来得换一个方法使用Parameter了

你可能感兴趣的:(python,深度学习)