ONNX GPT-2 モデルの変換

危険

ここで説明されているコードは非推奨になりました。従来のソリューションの適用を避けるため使用しないでください。下位互換性を確保するためにしばらく保持されますが、最新のアプリケーションでは使用してはなりません。

このガイドでは、非推奨となった変換方法について説明します。新しい推奨方法に関するガイドは、Python チュートリアルに記載されています。

公開されている事前トレーニング済み GPT-2 モデルは、テキスト内の前の単語がすべて与えられると、次の単語を予測するという単純な目的の大規模トランスフォーマー・ベースの言語モデルです。

事前トレーニングされたベース GPT-2 モデルをダウンロード

モデルをダウンロードするには、このモデルに移動し、[Download] を押します。

モデルとサンプル・テスト・データをダウンロードするには、このモデルに移動し、[Download] をクリックします。

ONNX GPT-2 モデルから IR への変換

次のパラメーターを使用してモデル変換を行い、GPT-2 モデルの中間表現を生成します。

mo --input_model gpt2-10.onnx --input_shape [X,Y,Z] --output_dir <OUTPUT_MODEL_DIR>