Failed to save model graph

[qgao21]

Hello author,
Thank you very much for your code, I have encountered a problem in use, I installed all the packages as required, the code can be trained normally, but the error is reported’Failed to save model graph:‘.I can't find a solution and hope the author can guide me。

[lidq92]

@gq123smu It is an onnx problem since the CNNIQA model contains some incompatible function arguments. If this model is replaced with normal operations (such as conv, max pooling, fc, etc), e.g., AlexNet, the error will go away.

It doesn't matter, this part is only for network visualization. It will not affect the training and testing processes.

[qgao21]

Thank you for your timely answer.