Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

to onnx #10

Open
dxw2000 opened this issue Oct 13, 2023 · 0 comments
Open

to onnx #10

dxw2000 opened this issue Oct 13, 2023 · 0 comments

Comments

@dxw2000
Copy link

dxw2000 commented Oct 13, 2023

Hello author, may I ask if your model can transfer onnx? There will be an error when I use this model to transfer onnx
Traceback (most recent call last):
File "To_onnx.py", line 65, in
torch.onnx.export(model, x, 'best.onnx', input_names=input_names, opset_version=15, output_names=output_names, verbose='False')
File "/home/iv/miniconda3/envs/dxw/lib/python3.8/site-packages/torch/onnx/init.py", line 350, in export
return utils.export(
File "/home/iv/miniconda3/envs/dxw/lib/python3.8/site-packages/torch/onnx/utils.py", line 163, in export
_export(
File "/home/iv/miniconda3/envs/dxw/lib/python3.8/site-packages/torch/onnx/utils.py", line 1074, in _export
graph, params_dict, torch_out = _model_to_graph(
File "/home/iv/miniconda3/envs/dxw/lib/python3.8/site-packages/torch/onnx/utils.py", line 731, in _model_to_graph
graph = _optimize_graph(
File "/home/iv/miniconda3/envs/dxw/lib/python3.8/site-packages/torch/onnx/utils.py", line 308, in _optimize_graph
graph = _C._jit_pass_onnx(graph, operator_export_type)
File "/home/iv/miniconda3/envs/dxw/lib/python3.8/site-packages/torch/onnx/init.py", line 416, in _run_symbolic_function
return utils._run_symbolic_function(*args, **kwargs)
File "/home/iv/miniconda3/envs/dxw/lib/python3.8/site-packages/torch/onnx/utils.py", line 1406, in _run_symbolic_function
return symbolic_fn(g, *inputs, **attrs)
File "/home/iv/miniconda3/envs/dxw/lib/python3.8/site-packages/torch/onnx/symbolic_helper.py", line 219, in wrapper
args = [
File "/home/iv/miniconda3/envs/dxw/lib/python3.8/site-packages/torch/onnx/symbolic_helper.py", line 220, in
_parse_arg(arg, arg_desc, arg_name, fn_name) # type: ignore[assignment]
File "/home/iv/miniconda3/envs/dxw/lib/python3.8/site-packages/torch/onnx/symbolic_helper.py", line 97, in _parse_arg
raise RuntimeError(
RuntimeError: Failed to export an ONNX attribute 'onnx::Cast', since it's not constant, please try to make things (e.g., kernel size) static if possible

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant