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'
