核 Python API - 3.5 简体中文

Vitis AI 用户指南 (UG1414)

Document ID
UG1414
Release Date
2023-09-28
Version
3.5 简体中文
表 1. get_target_info()
描述 参数 返回
获取目标信息,包括批次、指纹和目标名称。此信息可用于批处理或者获取目标名称信息。 DeviceInfo 对象。如需了解有关此对象类型的更多详情,请参阅核的各类部分。
表 2. create_wego_graph(input_graph_def, feed_dict={}, accuracy_mode= vitis_vai.enums.AccuracyMode.Default)
描述 参数 返回
用于 VAI 变换的 Python 封装文件。
  1. input_graph_def:包含要变换的模型的 GraphDef 对象。
  2. feed_dict:输入不含固定输入形状的模型时,用于推断形状配置。
  3. accuracy_mode:
    • vitis_vai.enums.AccuracyMode.Default:在不含 CPU FixNeuron 的情况下运行。
    • vitis_vai.enums.AccuracyMode.ReserveReduantFixNeurons:在含 CPU FixNeruon 的情况下运行
新 GraphDef,其中所含 VaiWeGOOp 置于计算图中以替代子计算图。
注释: 默认情况下,WeGO 通过消除量化模型中的 CPU FixNeuron 运算符来达成最优性能。但对于包含大量 CPU FixNeuron 运算符的模型,以默认值 (Vitis_vai.enums.AccuracyMode.Default) 部署这些模型的精度时,其精度可能精度降低。在此类情况下,您可切换至 Vitis_vai.enums.AccuracyMode.ReserveReduantFixNeurons 以提升精度。