次の表に、Vitis AI プルーナーがサポートする機能をフレームワーク別に示します。
フレームワーク | バージョン | 機能 | ||
---|---|---|---|---|
反復 | ワンステップ | OFA | ||
PyTorch | 1.4 ~ 1.10 をサポート | あり | あり | あり |
TensorFlow 1.15 | 1.15 をサポート | あり | なし | なし |
TensorFlow 2.x | 2.3 ~ 2.8.0 をサポート | あり | なし | なし |
注記:
Vitis AI 2.5 以降のリリースでは、Caffe および Darknet はサポートされません。これら 2 つのフレームワークで Vitis AI プルーナーを使用するには、それ以前のバージョンの Vitis AI を使用してください。