Сохранение и загрузка контрольной точки модели Pytorch для вывода не работает
Я думаю, что вы ищете это:
getNewItems: function (order) {
return db.ref('orders').orderByChild('deliveryDuration').equalTo(45));
}}
Этот запрос Firebase принимает каждый дочерний узел запрашиваемого местоположения, заказывает его по свойству deliveryDuration, а затем фильтрует только для получения те, которые равны 45.
1 ответ
Здесь следует учитывать две вещи.
-
Вы упомянули, что вы тренируете свою модель на GPU и используете ее для вывода на CPU, поэтому вам нужно добавить параметр map_location в load передача функции torch.device ('cpu') .
-
Существует несоответствие ключей state_dict (указано в вашем выходном сообщении), которое может быть вызвано некоторыми отсутствующими ключами или наличием большего количества ключей в state_dict , который вы загружаете, чем используемой вами моделью В настоящее время. И для этого вы должны добавить параметр strict со значением False в функции load_state_dict . Это заставит метод игнорировать несоответствие ключей.
Примечание: попробуйте использовать расширение pt или pth для файлов контрольных точек, как это принято.