ONNX

PyTorchモデルをONNX化する際にカスタムオペレータの実装が必要っぽい件

やりたかったこと PyTorchモデルをONNXファイルにexportしたい モデルにはカスタムオペレータを含む カスタムオペレータの実装は存在しない(該当部分は推論不能) カスタムオペレータはONNXでそれらしきop_typeを持つノードで出力してくれればOK やってみた…

pip installできるようにしました

前回作ったなんちゃってライブラリを改良?したので報告。 maminus.hatenadiary.org 前回はライブラリをpip install .すらできなかった。 こんなエラーになる。 $ pip install . Processing /home/maminus/python3.6/ommle ERROR: Command errored out with …

ONNXモデルの中間ノード出力を抽出するライブラリを作ってみた

先日記事にしたこれを実装してみた。 maminus.hatenadiary.org ソースコード GitHubに上げました。まともな?ライブラリを作るの初めてなので何か間違ってるかも・・・(特にPipenv自信なし。pip install も直接gitを指定できなかった。たぶんどこか間違って…