我想知道是否可以使用轉換為 onnx 模型的 tensorflow 和 pytorch 模型來使用 C Api 訓練它們,就像在例如https://gist.github.com/asimshankar/5c96acd1280507940bad9083370fe8dc 中使用 tensorflow 模型完成的那樣。我剛剛找到了使用 onnx 進行推理的示例。這個想法是能夠在 python 中使用 tensorflow 和 pytorch 進行原型設計,轉換為 onnx 模型并在 C 中有一個統一的 API 來進行推理和訓練。獲取一些(獲取鏈接)資訊會很有幫助。
uj5u.com熱心網友回復:
ONNX的 GitHub 頁面表明它可以用于推理,但能夠用它訓練所有模型似乎并不合理(從開發角度來看)。
目前,我們專注于推理(評分)所需的功能。
雖然有一些困難,比如總是寫backpropagation總比困難多feedforwarding,并支持其將增加一倍的框架大小,這是不是有什么ONNX的目標是因為已經有這這么多的框架。訓練你需要所有的引數,GPU 和 CPU 中函式的導數(如果它的性能低于其他框架,那將是一個大問題,因為沒有人會使用它)。還有許多其他因素使統一框架變得困難(例如,支持通過網路在多個 GPU 上進行訓練)。(所以在我們看來,這很好,但在他們看來卻如此困難)
但是我們可以看到框架中添加了一些訓練功能,在這種情況下它可以訓練變壓器模型
此外,要在PyTorch您中訓練變壓器,您可以查看此鏈接
uj5u.com熱心網友回復:
ONNX Runtime 支持訓練,但不支持 C 。您可以使用 ORT 和 Pytorch 訓練 ONNX 模型。請參閱此處https://onnxruntime.ai/docs/get-started/training-pytorch.html。
轉載請註明出處,本文鏈接:https://www.uj5u.com/yidong/350054.html
上一篇:為什么函式多載不提示錯誤?
