When loading the training model during the test, an error was reported that the model did not match

[QQuince]

Hi！
The model MFSNet.pth saved by training does not match when it is loaded in the test, and then I find that some parts of train network and test network are different. May I ask how you apply the trained model to the test?Looking forward to your reply！Thank you very much!
Best,
Quince.

[Luanatabea]

Hi! Did you solve the issue?