エントリーの編集
エントリーの編集は全ユーザーに共通の機能です。
必ずガイドラインを一読の上ご利用ください。
PyTorch トレーニング モデルを ONNX に変換する
記事へのコメント0件
- 注目コメント
- 新着コメント
このエントリーにコメントしてみましょう。
注目コメント算出アルゴリズムの一部にLINEヤフー株式会社の「建設的コメント順位付けモデルAPI」を使用しています
- バナー広告なし
- ミュート機能あり
- ダークモード搭載
関連記事
PyTorch トレーニング モデルを ONNX に変換する
このチュートリアルの前の段階では、PyTorch を使用して機械学習モデルを作成しました。 ただし、そのモ... このチュートリアルの前の段階では、PyTorch を使用して機械学習モデルを作成しました。 ただし、そのモデルは、.pth ファイルです。 Windows ML アプリと統合するには、モデルを ONNX 形式に変換する必要があります。 モデルのエクスポート モデルをエクスポートするには、torch.onnx.export() 関数を使用します。 この関数によってモデルを実行し、出力を計算するために使用される演算子のトレースを記録します。 Visual Studio で、次のコードを PyTorchTraining.py ファイルの main 関数の上にコピーします。 import torch.onnx #Function to Convert to ONNX def Convert_ONNX(): # set the model to inference mode model.eval()