工具与软件：
你(们)好
我尝试在 edgeai-modeloptimization r10.0上测试 QAT。
"我已经使用并参考 edgeai-torchmodelopt.xmodelopt.quantization v2和 QAT 指南、完成了针对经过训练的深度学习模型的 QAT (量化感知培训)。 (https://github.com/TexasInstruments/edgeai-tensorlab/blob/r10.0/edgeai-modeloptimization/torchmodelopt/edgeai_torchmodelopt/xmodelopt/quantization/v2/docs/qat.md)
但是、我在将 QAT 模型转换为 ONNX 的过程中遇到了一个错误。
我搜索了相关材料、但信息不足、无法解决问题。"
当我尝试将火炬模型转换为 ONNX 时、出现了 Attribute 错误。
如何解决？
这是我用于转换的代码。
它是转换的模型。
这是错误日志。
回溯(最近的呼叫最后)：
export_torch2onnx 中的文件"/home/bjkim/works/DL/torchmodelopt_test/scripts/export_onnx.py "、第128行
torch.onnx.export (模型、
导出文件/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/onnx/utils.py、第516行
_export(
在_export 中添加了文件"/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/onnx/utils.py、第1612行
graph、params_dict、torch_out =_model_to_graph (
在_model_to_graph 中、文件"/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/onnx/utils.py、第1133行
模型=_pre_trace_quant_model (模型、参数)
pre_trace_quant_model 中的文件"/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/onnx/utils.py、第1088行
return TORCH.JIT.TRACE (model、args)
文件"/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/jit/_trace.py "、行820、处于跟踪状态
返回 trace_module (
trace_module 中、文件"trace_module"/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/jit/_trace.py、第1053行
模块= make_module (mod、_module_class、_compilation_unit)
在 make_module 中、文件"make_module"、第630行、为/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/jit/_trace.py
返回_module_class (mod、_compilation_unit=_compilation_unit)
文件"1203"/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/jit/_trace.py、第1203行、位于中 初始化
tmp_module._modules[name]= make_module(.
在 make_module 中、文件"make_module"、第630行、为/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/jit/_trace.py
返回_module_class (mod、_compilation_unit=_compilation_unit)
文件"1203"/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/jit/_trace.py、第1203行、位于中 初始化
tmp_module._modules[name]= make_module(.
在 make_module 中、文件"make_module"、第630行、为/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/jit/_trace.py
返回_module_class (mod、_compilation_unit=_compilation_unit)
文件"1203"/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/jit/_trace.py、第1203行、位于中 初始化
tmp_module._modules[name]= make_module(.
在 make_module 中、文件"make_module"、第630行、为/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/jit/_trace.py
返回_module_class (mod、_compilation_unit=_compilation_unit)
文件"1203"/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/jit/_trace.py、第1203行、位于中 初始化
tmp_module._modules[name]= make_module(.
在 make_module 中、文件"make"/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/jit/_trace.py、第624行
返回 TORCH.JIT._RECURSIVE.CREATE_SCRIPT_MODULE (
create_script_module 中的文件"script_module"/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/jit/_recursive.py 第558行
返回 create_script_module_impl (nn_module、con要 类型、stubs_fn)
文件"script_module_impl"中的第573行"/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/jit/_recursive.py "
HOOK_STUBS、PRE_HOOK_STUBS = GET_HOOK_STUBS (nn_MODULE)
get_hook_stubs 中的文件"get"/home/bjkim/anaconda3/envs/pytorch230/lib/python3.10/site-packages/torch/jit/_recursive.py 第909行
如果挂钩。 文件夹 在 HOOK_MAP 中：
AttributeError:'functools.partial'对象没有属性' 文件夹 '。 您是指：" NE '？
这是我的环境
edgeai-tidl-tools 版本是 rel_10.00
Protobuf=3.20.3.
TORCH=2.3.0+cu118
onnx=1.14.0
