ONNX 运行时上的反向传播

Question

我想用 C++ 在 ONNX 模型上实现反向传播。这是过程，我在 pytorch 中训练模型，然后将其导出到 ONNX。我使用 ONNX 运行时导入模型，现在我的目标是在我导入的这个模型上实现反向传播。我是 ONNX 的新手，尝试在互联网上搜索，但找不到任何有用的东西。

简单来说，我想在 GPU 上运行反向传播。

我搜索了有关训练的 ONNX 运行时文档，但它没有提供任何与反向传播相关的信息。

如果有人可以就这个问题提供一些指导，我会很高兴。

Answer 1

在互联网上搜索一段时间后，我在 Github 上发现了一篇帖子，它让我了解了如何更新 ONNX 模型的权重。下面是链接。

https://github.com/onnx/onnx/issues/2978

从 ONNX 模型获取权重，提供输入数据，获得预测，然后实施反向传播并计算更新的权重，最后替换它们保存模型。