Pytorchs的nn.Linear在某些条件下仅返回`nan`值

时间:2019-03-30 18:37:55

标签: python machine-learning pytorch

我有一个pytorch张量,其规范化数据有时会通过模型的这一层,

linear = nn.Linear(in_features=137, out_features=1)

可以在此处查看数据

enter image description here

enter image description here

问题是结果我只得到nan

enter image description here

并且linear的参数也似乎是nan

enter image description here

为了希望更好地理解问题所在,我在那个确切的时刻将数据保存到了csv文件中,并将其加载到了解释器中,

enter image description here

执行精确的操作会产生预期的结果,

>>> linear(x)
tensor([[ 0.0594],
        [ 0.1051],
        [ 0.1000],
        ...,
        [ 0.1043],
        [-0.1210],
        [-0.0332]], grad_fn=<AddmmBackward>)

这种奇怪的行为可以归因于何处,我该如何解决?

0 个答案:

没有答案