asyaevloeva
@asyaevloeva

Как сохранить модель на Pytorch аналогично сохранению модели в Tensorflow?

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

model = keras.models.load_model('path/to/location')
prediction = model.predict(my_new_record)

но когда я делаю то же самое на пайторче:

model = torch.load('model.pt')
prediction = model(my_new_record)

то получаю:
TypeError: ‘collections.OrderedDict’ object is not callable

или
model.eval()
то
AttributeError: 'collections.OrderedDict' object has no attribute 'eval'


Как сохранять модель на пайторче таким образом, чтобы потом не нужно было снова инициализировать TheModelClass, а можно было как в тензорфлоу любую модель сохранить и потом просто загрузить ее и начать использовать ?
  • Вопрос задан
  • 43 просмотра
Пригласить эксперта
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы