python tools/export_onnx.py models/efficientnetv2/efficientnetv2_l.py --checkpoint datas/efficientnetv2-l_in21k-pre-
3rdparty_in1k_20221220-63df0efd.pth
Initialize the weights.
The model and loaded state dict do not match exactly
size mismatch for head.fc.weight: copying a param with shape torch.Size([1000, 1280]) from checkpoint, the shape in current model is torch.Size([5, 1280]).
size mismatch for head.fc.bias: copying a param with shape torch.Size([1000]) from checkpoint, the shape in current model is torch.Size([5]).
Exporting onnx model.
onnx_path datas/efficientnetv2-l_in21k-pre-3rdparty_in1k_20221220-63df0efd.onnx
Traceback (most recent call last):
File "tools/export_onnx.py", line 74, in <module>
main()
File "tools/export_onnx.py", line 65, in main
do_constant_folding=True
File "/opt/miniconda3/envs/Awesome-Backbones/lib/python3.6/site-packages/torch/onnx/__init__.py", line 230, in export
custom_opsets, enable_onnx_checker, use_external_data_format)
File "/opt/miniconda3/envs/Awesome-Backbones/lib/python3.6/site-packages/torch/onnx/utils.py", line 91, in export
use_external_data_format=use_external_data_format)
File "/opt/miniconda3/envs/Awesome-Backbones/lib/python3.6/site-packages/torch/onnx/utils.py", line 639, in _export
dynamic_axes=dynamic_axes)
File "/opt/miniconda3/envs/Awesome-Backbones/lib/python3.6/site-packages/torch/onnx/utils.py", line 411, in _model_to_graph
use_new_jit_passes)
File "/opt/miniconda3/envs/Awesome-Backbones/lib/python3.6/site-packages/torch/onnx/utils.py", line 379, in _create_jit_graph
graph, torch_out = _trace_and_get_graph_from_model(model, args)
File "/opt/miniconda3/envs/Awesome-Backbones/lib/python3.6/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 "/opt/miniconda3/envs/Awesome-Backbones/lib/python3.6/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 "/opt/miniconda3/envs/Awesome-Backbones/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/opt/miniconda3/envs/Awesome-Backbones/lib/python3.6/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
错误的信息如下：