мне нужно сохранить модель пайторча для последующего предсказания на новых данных, чтобы загрузить модель и использовать можно было аналогично тому как это в тензорфлоу:
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, а можно было как в тензорфлоу любую модель сохранить и потом просто загрузить ее и начать использовать ?
Простой
