pytorch_nndct.get_pruning_runner - 2.5 简体中文

Vitis AI 优化器用户指南 (UG1333)

Document ID
UG1333
Release Date
2022-06-15
Version
2.5 简体中文

此 API 具有下列方法:

get_pruning_runner(model, inputs, method)
model
要进行剪枝的 torch.nn.Module 对象。
inputs
单个 torch.Tensor 或列表,用作为模型推断的输入。输入无需采用真实数据。可以采用随机生成的张量,与真实数据的形状和数据类型相同即可。
method
可取“iterative”或“one_step”。