PyTorch→ONNX→NNVM - Qiita

はじめに こちらの記事で紹介したNNVMですが、記事内であげていた OpenCLビルドが通らない PyTorchからのONNX exportが通らない という問題は開発が進み解消されましたので、その分を書きます。 今回はPyTorchの学習済モデルをONNX経由でLLVM/OpenCLビルドして、PyTorchと実行速度を比較してみます。OpenCLのターゲットは MacBook Pro のIntel HD Graphics4000です。 学習済モデルの確認 torch-visionの学習済モデルを使います。VGG,Alexnet,Resnet,Squeezenet,Inception,Densenetが用意されています。下記のように簡単に学習済モデルが得られます。