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

Question

[Ася]

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

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