Существует ли удобный способ сохранять информацию об архитектуре модели в Pytorch только в файле с прототипом? - PullRequest
1 голос
/ 07 апреля 2020

Существует ли удобный способ сохранять информацию об архитектуре модели в Pytorch только в файле протобуфа с правами?

Я знаю, как использовать pytorch.save для сохранения обоих весов и net одновременно, чтобы словарь структурированных данных. Но если я хотел бы сохранить данные в изолированном файле, который содержит только архитектуру net, как, например, то, что Каффе тренировал из исходного состояния, это возможно? Файл может использоваться где-то еще. Может ли ONNX сделать что-то подобное?

...