我在导出onnx模型遇到这个问题

[oowwkk]

File "site-packages\omegaconf\dictconfig.py", line 480, in _get_node
raise ConfigKeyError(f"Missing key {key!s}")
omegaconf.errors.ConfigAttributeError: Missing key load_state_dict
full_key: model.load_state_dict
object_type=dict