专栏算法工具链hbm模型计算量获取方式

hbm模型计算量获取方式

已解决
默认383362025-04-10
41
5

请问qat量化得到的hbm模型有什么方式可以获得模型计算量？类似于图片中ptq量化工具hb_perf输出结果中的信息

附件:
算法工具链
征程5
评论4
0/1000
  • Huanghui
    Lv.5

    你好， 目前计算量工具支持统计计算量的op有三种，分别为torch.nn.Conv2d，torch.nn.Linear 以及 torch.nn.ConvTranspose2d

    2025-04-10
    0
    1
    • 默认38336回复Huanghui:

      好的谢谢！

      2025-04-10
      0
  • Huanghui
    Lv.5

    其中config中影响计算量的主要key是test_model（或者model），以及test_inputs。model相关的决定了计算量工具需要检查的模型，而test_inputs决定了输入的大小，除此之外，输入形状还可以通过--input-shape B,C,H,W的输入参数决定。

    2025-04-10
    0
    0
  • DR_KAN
    Lv.4

    qat也可以得到性能分析报告，有专门的python api，可以看下手册的介绍

    2025-04-10
    0
    0
  • 费小财
    Lv.5

    你看看这个帖子 有没有相关的信息https://developer.horizon.auto/blog/12895

    2025-04-11
    0
    0