Pytorch Ошибка при загрузке модели

17.12.2021, 20:21. Показов 1027. Ответов 0

Метки нет (Все метки)

Студворк — интернет-сервис помощи студентам

Прошу не кидаться камнями, я недавно начал изучать Python) Нашел интересную модель для классификации изображений, обучил, все хорошо работает. Но я не могу в новом файле использовать эту уже обученную модель. То есть:
В первом файле после обучения сохранял так: torch.save(model.state_dict,'/content/drive/MyDrive/NN/results.pth')

Python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
from __future__ import print_function, division
import torch
import torch.nn as nn
import torch.optim as optim
from torch.optim import lr_scheduler
import torchvision
from torchvision import datasets, models, transforms
import matplotlib.pyplot as plt
import numpy as np
import time
import os
import copy
# Путь к датасету
data_dir = '/content/drive/MyDrive/NN'
 
# Создание датасетов с помощью Image Folder
chosen_datasets = {x: datasets.ImageFolder(os.path.join(data_dir, x),
  chosen_transforms[x])
                  for x in ['train', 'val']}
 
# Загрузка данных
dataloaders = {x: torch.utils.data.DataLoader(chosen_datasets[x], batch_size=5,
  shuffle=True, num_workers=4)
              for x in ['train', 'val']}
 
dataset_sizes = {x: len(chosen_datasets[x]) for x in ['train', 'val']}
class_names = chosen_datasets['val'].classes
 
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
# Загружаем модель
model = torch.load('/content/drive/MyDrive/NN/results.pth')
print (model)
Print (inputs)
outputs = model(inputs)

Выдает ошибку: 'Tensor' object has no attribute '_metadata'. Outputs нужны для отображения результатов работы НС. В первом все работает с таким синтаксисом сразу после обучения модели. В чем проблема? Model и Inputs загружены нормально. Файл модели весит больше 100 МБ. Может быть проблема в том, что я Google Collab использую, так как удобно с разных компов, да и библиотеки не ставить?

Programming

Эксперт

39485 / 9562 / 3019

Регистрация: 12.04.2006

Сообщений: 41,671

Блог

17.12.2021, 20:21

Ответы с готовыми решениями:

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

Ошибка при сохранении и загрузке конфигурации УТП при загрузке файл конфигурации cf до версии 1.2.66
Здравствуйте, ошибка при сохранении и загрузке конфиурации. Была типичная конфигурация с данными "Управление торговым...

При установке пакета pytorch выдает ошибку
С сайта взял ссылку и ввел команду pip3 install https://download.pytorch.org/whl/cpu/torch-1. 1.0-cp37-cp37m-win_amd64.whl , при этом...

Надоела реклама? Зарегистрируйтесь и она исчезнет полностью.

inter-admin

Эксперт

29715 / 6470 / 2152

Регистрация: 06.03.2009

Сообщений: 28,500

Блог

17.12.2021, 20:21

Помогаю со студенческими работами здесь

imageio.imread в коде нейронной сети выдает ошибку, работая с pytorch, не хочет работать, где ошибка!
Всем доброго времени суток! Случилась вот какая беда! Работаю над магисторской выпускной работой , надо запустить нейронную сеть...

Собрал компьютер, вылетает ошибка: ошибка при загрузке AMBspi.dll ,не находит указаный модуль
Собрал комп для игр. Материнка Crosshair 4 formula, графика ASUS HD 6870; процесор AMD X4 965, 3,4 GHz, Инсталировал вроде как всё...

Логическая ошибка при выполнении простой модели при работе с очередями с использованием приоритетов
Здравствуйте уважаемые форумчане! работаю в Anylogic 7.0. При построении простой системы массового обслуживания, с использованием...

Ошибка ssis пакета (vs2008) при загрузке .csv файла в БД при запуске VS2015
SSIS пакет создан в vs2008. Работал. Сменился MS Server на 2015 и vs2015. Загружаю. Выдает ошибку ] Ошибка: Ошибка при обработке...

Ошибка при загрузке фото - нарушение прав доступа при чтении по адресу 0x00814000
При попытке компиляции кода выскакивает ошибка: Первый этап обработки исключения по адресу 0x5E4FF20C (msvcr120.dll) в TestSFML.exe:...

Искать еще темы с ответами

Или воспользуйтесь поиском по форуму:

Новые блоги и статьи Все статьи Все блоги /
Уведомление о неверно выбранном значении справочника Maks 06.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "НарядПутевка", разработанного в конфигурации КА2. Задача: уведомлять пользователя, если в документе выбран неверный склад. . .	Установка Qt Creator для C и C++: ставим среду, CMake и MinGW без фреймворка Qt 8Observer8 05.04.2026 Среду разработки Qt Creator можно установить без фреймворка Qt. Есть отдельный репозиторий для этой среды: https:/ / github. com/ qt-creator/ qt-creator, где можно скачать установщик, на вкладке Releases:. . .	AkelPad-скрипты, структуры, и немного лирики.. testuser2 05.04.2026 Такая программа, как AkelPad существует уже давно, и также давно существуют скрипты под нее. Тем не менее, прога живет, периодически что-то не спеша дополняется, улучшается. Что меня в первую очередь. . .	Отображение реквизитов в документе по условию и контроль их заполнения Maks 04.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "ПланированиеСпецтехники", разработанного в конфигурации КА2. Данный документ берёт данные из другого нетипового документа. . .
Фото всей Земли с борта корабля Orion миссии Artemis II kumehtar 04.04.2026 Это первое подобное фото сделанное человеком за 50 лет. Снимок называют новым вариантом легендарной фотографии «The Blue Marble» 1972 года, сделанной с борта корабля «Аполлон-17». Новое фото. . .	Вывод диалогового окна перед закрытием, если документ не проведён Maks 04.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "СписаниеМатериалов", разработанного в конфигурации КА2. Задача: реализовать программный контроль на предмет проведения документа. . .	Программный контроль заполнения реквизитов табличной части документа Maks 02.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "СписаниеМатериалов", разработанного в конфигурации КА2. Задача: 1. Реализовать контроль заполнения реквизита. . .	wmic не является внутренней или внешней командой Maks 02.04.2026 Решение: DISM / Online / Add-Capability / CapabilityName:WMIC~~~~ Отсюда: https:/ / winitpro. ru/ index. php/ 2025/ 02/ 14/ komanda-wmic-ne-naydena/

Наверх