Stage2 pretrained weights key mismatch with CSDLLSRNetv9_7_5 (UltraIS)

[casual111111]

Hi maintainers,

I’m unable to load the provided Stage2 pretrained weights with the model CSDLLSRNetv9_7_5.

What I did

Use config: Super_Resolution/Options/CSDLLSR_v9_7_5_3_scale4_test.yml

Set:

pretrain_network_g: pretrain/stage2/stage2_weight.pth.tar

Run:

sh test.sh Super_Resolution/Options/CSDLLSR_v9_7_5_3_scale4_test.yml

What I found

stage2_weight.pth.tar contains only these components (480 keys total):

module.Unet.* (332 keys)

module.decom.* (148 keys)

But the model CSDLLSRNetv9_7_5 expects parameters under:

illumination.*

reflectance.*

seg.*

So loading fails due to missing/unexpected keys. Even with strict_load_g: False, only a subset loads and the rest remains uninitialized.

Question

Could you please share the correct Stage2 pretrained weights for CSDLLSRNetv9_7_5, or clarify which config/model version stage2_weight.pth.tar is intended for (and how we should load it)?

Thanks!