The following tables shows the features that are supported by the Vitis AI pruner for different frameworks:
|PyTorch||Supports 1.4 - 1.10||Yes||Yes||Yes|
|TensorFlow 1.15||Supports 1.15||Yes||No||No|
|TensorFlow 2.x||Supports 2.3 - 2.8.0||Yes||No||No|
Note: Vitis AI 2.5 and later releases do not support Caffe and Darknet. To use the Vitis AI Pruner on these two frameworks, use an earlier version of Vitis AI.