Как сохранить и использовать обученную нейронную сеть, разработанную в PyTorch / TensorFlow / Keras?

Question

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

Answer 1

PyTorch: Как объяснено в этом посте , вы можете сохранить параметры модели в виде словаря или загрузить словарь, чтобы установить параметры вашей модели.Вы также можете сохранить / загрузить модель PyTorch как объект.Обе процедуры требуют, чтобы у пользователя была установлена ​​хотя бы одна структура тензорных вычислений, например, для эффективного умножения матриц.