このドキュメントは AI によって生成されました。誤りを発見された場合、または改善のご提案がありましたら、ぜひご貢献ください！ GitHub で編集するTorchCompileModel ノードは、PyTorch のコンパイル機能をモデルに適用し、そのパフォーマンスを最適化します。このノードは入力モデルのコピーを作成し、指定されたバックエンドを用いて PyTorch のコンパイル機能でラップします。これにより、推論時のモデル実行速度が向上します。
入力
| パラメーター | データ型 | 必須 | 範囲 | 説明 |
|---|---|---|---|---|
model | MODEL | はい | - | コンパイルおよび最適化対象のモデル |
backend | STRING | はい | ”inductor" "cudagraphs” | 最適化に使用する PyTorch コンパイルバックエンド |
出力
| 出力名 | データ型 | 説明 |
|---|---|---|
model | MODEL | PyTorch コンパイルが適用されたコンパイル済みモデル |