保存和加载未序列化的 pytorch 预训练模型

Question

对于正常的pytorch模型的保存和加载非常简单。

在python中我可以使用torch.save(model, FilePath)和c++torch::jit::load(FilePath)。并且可以将保存的模型和c++加载代码放在一个目录中。但是有一个限制，不能在生产模式下目录中包含二进制文件（请不要问我为什么不能包含二进制文件，我也很疑惑）。

所以，我想知道如何在没有序列化的情况下从python保存pytorch模型并在c++中加载这个模型。可能吗？

Answer 1

使用 ONNX 文件格式。 Pytorch 可以输出这种格式的文件，然后你应该可以找到相关的库在 C++ 中加载它。