KeyError on Loading SPSL Detector with xception_best.pth

[benliang99]

I'm encountering an issue when attempting to load the SPSL detector model using the xception_best.pth file. During the initialization, the following error is thrown:

KeyError: 'conv1.weight'

It appears that the model is expecting a key that doesn't exist in the provided state dictionary. I suspect that the correct key might be 'backbone.conv1.weight'. Can someone confirm if this is the case, or suggest a potential fix if the key naming has changed in recent updates?

Environment Details:
Framework (e.g., PyTorch) Version: 2.4.0+cu121
Platform: Ubuntu

Thank you for any assistance you can provide!