Error when exporting the model to ONNX

[nitrif]

I get following error when trying to export the model to ONNX format once the pretrained model is loaded, by adding torch.onnx.export(...) command. Do you know what might be the cause of this?

Thanks, Nikola

Message=Failed to export an ONNX attribute, since it's not constant, please try to make things (e.g., kernel size) static if possible
Source=
StackTrace:
File "E:\git\hrnet-image-classification\tools\valid.py", line 100, in main
torch.onnx.export(model, dump_input, r'e:\hrnetv2_w18_imagenet_pretrained.onnx', verbose=True)
File "E:\git\hrnet-image-classification\tools\valid.py", line 136, in
main()