专栏算法工具链qat模型非图像输入如何不量化使用原始浮点数据输入

qat模型非图像输入如何不量化使用原始浮点数据输入

已解决
yy12342024-12-26
52
1

PTQ量化时非图像输入可以配置成特征图输入,float数据类型即可,qat训练自带示例代码转换出来的模型的非图像输入都是下图类似的s16或s8,因此需要对输入的文件进行量化操作。请问qat可以输入浮点数吗?我们想和ptq一样输入浮点数据,在模型中完成量化,不想在外面手动处理。不知道qat训练在哪里配置这个。

 

 

 

算法工具链
征程5
评论1
0/1000
  • Huanghui
    Lv.5

    你好,排查了一下工具链的qat流程中用到的脚本,示例中之所以input是整型的,应该编译脚本

    /open_explorer/samples/ai_toolchain/horizon_model_train_sample/scripts/tools/compile_perf_hbir.py中执行 remove_quant_dequant(self.model) 的原因,你可以执行他们就OK了:

    2025-01-10
    0
    0