我有一个pytorch
张量,其规范化数据有时会通过模型的这一层,
linear = nn.Linear(in_features=137, out_features=1)
可以在此处查看数据
问题是结果我只得到nan
并且linear
的参数也似乎是nan
,
为了希望更好地理解问题所在,我在那个确切的时刻将数据保存到了csv文件中,并将其加载到了解释器中,
执行精确的操作会产生预期的结果,
>>> linear(x)
tensor([[ 0.0594],
[ 0.1051],
[ 0.1000],
...,
[ 0.1043],
[-0.1210],
[-0.0332]], grad_fn=<AddmmBackward>)
这种奇怪的行为可以归因于何处,我该如何解决?