Nn.Linear Pytorchs возвращает только значения `nan` при определенных условиях - PullRequest
0 голосов
/ 30 марта 2019

У меня есть тензор pytorch с моими нормализованными данными, которые в какой-то момент проходят через этот слой в моей модели,

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

Данные можно увидеть здесь,

enter image description here

enter image description here

Проблема в том, что я получаю только nan s в результате

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>)

Куда можно отнести это странное поведение и как я могу это исправить?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...