使用export_ONNX时报错

[Maplexff]

(torch) PS D:\Awesome-Backbones-main> python tools/export_onnx.py models/efficientnetv2/efficientnetv2_m.py --checkpoint logs/EfficientNetV2/2025-04-06-19-59-56/Val_Epoch229-Acc98.525.pth
Initialize the weights.

Exporting onnx model.
Traceback (most recent call last):
File "tools/export_onnx.py", line 65, in
main()
File "tools/export_onnx.py", line 57, in main
do_constant_folding=True
File "D:\APP\anaconda3\envs\torch\lib\site-packages\torch\onnx_init_.py", line 230, in export
custom_opsets, enable_onnx_checker, use_external_data_format)
File "D:\APP\anaconda3\envs\torch\lib\site-packages\torch\onnx\utils.py", line 91, in export
use_external_data_format=use_external_data_format)
File "D:\APP\anaconda3\envs\torch\lib\site-packages\torch\onnx\utils.py", line 639, in _export
dynamic_axes=dynamic_axes)
File "D:\APP\anaconda3\envs\torch\lib\site-packages\torch\onnx\utils.py", line 411, in _model_to_graph
use_new_jit_passes)
File "D:\APP\anaconda3\envs\torch\lib\site-packages\torch\onnx\utils.py", line 379, in _create_jit_graph
graph, torch_out = _trace_and_get_graph_from_model(model, args)
File "D:\APP\anaconda3\envs\torch\lib\site-packages\torch\onnx\utils.py", line 342, in _trace_and_get_graph_from_model
torch.jit._get_trace_graph(model, args, strict=False, _force_outplace=False, _return_inputs_states=True)
File "D:\APP\anaconda3\envs\torch\lib\site-packages\torch\jit_trace.py", line 1148, in _get_trace_graph
outs = ONNXTracedModule(f, strict, _force_outplace, return_inputs, _return_inputs_states)(*args, **kwargs)
File "D:\APP\anaconda3\envs\torch\lib\site-packages\torch\nn\modules\module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "D:\APP\anaconda3\envs\torch\lib\site-packages\torch\jit_trace.py", line 93, in forward
in_vars, in_desc = _flatten(args)
RuntimeError: Only tuples, lists and Variables supported as JIT inputs/outputs. Dictionaries and strings are also accepted but their usage is not recommended. But got unsupported type bool