专栏算法工具链petr模型QAT训练时报错,不支持averaging_constant

petr模型QAT训练时报错,不支持averaging_constant

dlutguobin2023-08-10
45
1

OE版本1.1.57

J5板

报错信息如下:

File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/quantization/quantize.py", line 835, in _convert

reassign[name] = swap_module(mod, mapping, swapped_modules)

File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/quantization/quantize.py", line 760, in swap_module

new_mod = mapping[type(mod)].from_float(mod)

File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/nn/qat/softmax.py", line 119, in from_float

return SegmentLUTSoftmax.from_float(mod)

File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/nn/qat/softmax.py", line 378, in from_float

qat_softmax = cls(

File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/nn/qat/softmax.py", line 181, in __init__

self.sub = FloatFunctional(qconfig=sub_qconfig)

File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/nn/qat/functional_modules.py", line 24, in __init__

self.activation_post_process = qconfig.activation()

File "/root/.local/lib/python3.8/site-packages/torch/ao/quantization/observer.py", line 63, in __call__

return self.p(*args, **keywords)

File "/root/.local/lib/python3.8/site-packages/torch/ao/quantization/observer.py", line 63, in __call__

return self.p(*args, **keywords)

File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/utils/typeguard.py", line 1076, in wrapper

retval = func(*args, **kwargs)

File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/quantization/fake_quantize.py", line 101, in __init__

self.activation_post_process = observer(

TypeError: __init__() got an unexpected keyword argument 'averaging_constant'

算法工具链
评论1
0/1000
  • 颜值即正义
    Lv.2

    您好,此问题将在OE1.1.62版本解决,请关注版本发布帖https://developer.horizon.cc/forumDetail/118363912788935318

    2023-08-10
    0
    0