把模型转换为QAT状态时报错

1. 芯片型号：X3

报错信息如下，X3不支持包含leakyrelu的模型进行prepare_qat_fx吗？

Traceback (most recent call last):

 File "float_calib.py", line 59, in

  calib_model = prepare_qat_fx(

 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/quantize_fx.py", line 626, in prepare_qat_fx

  model = _prepare_fx(

 File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/quantization/quantize_fx.py", line 327, in _prepare_fx

  prepared = quantizer.prepare(

 File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/quantization/fx/quantize.py", line 517, in prepare

  return self._prepare(

 File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/quantization/fx/quantize.py", line 490, in _prepare

  self._qat_swap_modules(model, hybrid, opset_version)

 File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/quantization/fx/quantize.py", line 83, in _qat_swap_modules

  convert(

 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/quantize.py", line 766, in convert

  _convert(

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

  _convert(

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

  _convert(

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

  _convert(

 [Previous line repeated 1 more time]

 File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/quantization/quantize.py", line 962, 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 878, in swap_module

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

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

  qat_mod = cls(

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

  self.lut = SegmentLUT(

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

  assert self.activated(), "SegmentLUT only support march in {}!".format(

AssertionError: SegmentLUT only support march in ('bayes', 'bayes-e', 'meta', 'nash', 'nash-e', 'nash-m', 'nash-p')!