После обновления Pytorch моя модель выдает почти случайный вывод - PullRequest
0 голосов
/ 04 марта 2019

Я тренировался, тестировал и до сих пор использую модель в "Pytorch 0.4.1".Это было, и все еще работает нормально (вывод - то, что должно быть), если я использую шаг 0.4.1.Но когда я обновляюсь до версии 1.0.1, каждый раз, когда я пытаюсь оценить одно и то же входное изображение, я получаю разные выходные данные (это регрессия).

Я пытался увидеть, что изменилось в этих версиях, ноТак как я не получаю никаких ошибок и предупреждений, я не уверен, что мне следует искать конкретно.

PS: я проверил веса, они также совпадают, когда я загружаю модель

1 Ответ

0 голосов
/ 09 апреля 2019

Это было из-за выпадающего слоя.Model.eval отключает выпадающий слой.Довольно простоНо теперь в обновлении Pytorh, если Dropout не определен конкретно в функции модели init , он не отключится во время eval.По крайней мере, это было причиной для моего случая.

...