HVision-NKU/SRFormer

Real world model weight load problem about SRFormer_S_RealSR.pth

ZaizuoTang opened this issue · 1 comments

Thank you for your open source code, but when I load the pre-trained weights, some weights are lost.

pretrain_network_g: /home/tzz/Work_file/CODE/SRFormer-main/SRFormer_S_RealSR.pth
options: test_SRFormer-S_x4_real.yml

2024-04-25 21:51:18,810 INFO: Loading SRFormer model from /home/tzz/Work_file/CODE/SRFormer-main/SRFormer_S_RealSR.pth, with param key: [params_ema].
2024-04-25 21:51:18,834 WARNING: Current net - loaded net:
2024-04-25 21:51:18,834 WARNING: layers.0.residual_group.blocks.1.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.0.residual_group.blocks.3.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.0.residual_group.blocks.5.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.1.residual_group.blocks.1.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.1.residual_group.blocks.3.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.1.residual_group.blocks.5.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.2.residual_group.blocks.1.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.2.residual_group.blocks.3.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.2.residual_group.blocks.5.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.3.residual_group.blocks.1.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.3.residual_group.blocks.3.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.3.residual_group.blocks.5.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.4.residual_group.blocks.1.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.4.residual_group.blocks.3.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.4.residual_group.blocks.5.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.5.residual_group.blocks.1.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.5.residual_group.blocks.3.attn_mask
2024-04-25 21:51:18,834 WARNING: layers.5.residual_group.blocks.5.attn_mask
2024-04-25 21:51:18,834 WARNING: Loaded net - current net:
2024-04-25 21:51:18,859 INFO: Model [SRModel] is created.