Я обучил нейронную сеть с помощью Google Colab.
Я сохранил нейронную сеть, используя joblib.dump()
Затем я загрузил модель на свой P C, используя joblib.load()
Я сделал прогноз на одном и том же образце, используя ту же модель, и на colab, и на моем P C. На colab он имеет вывод [[0.51]]
. На моем p c он имеет вывод [[nan]]
.
Сводка модели сообщает, что архитектура модели такая же. Я проверил вес модели, которую я загрузил на свой P C, и модель на colab, и весы точно такие же.
Есть какие-нибудь идеи относительно того, что я могу сделать? Спасибо.
Быстрое обновление: даже если я изменю все свои входные данные на ноль, прогноз будет все еще nan.