Проблема с сохранением и загрузкой модели Pytorch ⇐ Python

[Anonymous]

Я пытаюсь разработать нейронную сеть для классификации сигналов с помощью модели Pytorch. после обучения модели я сохраняю модель и загружаю ее для этапа вывода.
Я пытаюсь разработать нейронную сеть для классификации сигналов с помощью модели Pytorch. после обучения модели я сохраняю модель с помощью следующего кода:

Код: Выделить всё

torch.save(model.state_dict(), ".pth")

после этого я хочу загрузить модель для этапа вывода. я использую следующий код:

Код: Выделить всё

model_test = spectrogram_model(X_Test)
model_test.load_state_dict(torch.load(".pth"))

но я получил следующую ошибку:

Код: Выделить всё

AttributeError: 'Tensor' object has no attribute 'load_state_dict'

у вас есть решение?

Подробнее здесь: https://stackoverflow.com/questions/782 ... orch-model