Hello, Author,
I successfully ran python train.py --project PROJECT_NAME --noisy-train-dir NOISY_IMAGE_TRAIN_DIR --clean-train-dir CLEAN_IMAGE_TRAIN_DIR --noisy-valid-dir NOISY_IMAGE_VALID_DIR --clean-valid-dir CLEAN_IMAGE_VALID_DIR and obtained the weight parameters SAR-CAM_best.pth. However, when I ran the second set of code for testing on the test set, I encountered a mismatch between the weights and the architecture when loading the weight parameters. The specific error code is:
RuntimeError: Error(s) in loading state_dict for SAR_CAM: Unexpected key(s) in state_dict: "conv_in.conv.bias", "conv_out.conv.bias", ... "up.up.1.conv_in.conv.bias", "up.bottleneck.conv.bias"...
I am not clear about the specific reason. I hope you can help to clarify. Thank you!