PyTorchとCaffe2でONNXを使ってみる

Deep Learning フレームワークざっくり紹介 Advent Calendar 2017 の 9日目 の記事です。 PyTorchとCaffe2で、モデル表現の標準フォーマットであるONNX (Open Neural Network Exchange)を使ってみます。 環境 PyTorch インストール モデル定義 Caffe2 ONNX インストール モデルのエクスポート モデルのインポート まとめ