Zj-BinXia / DiffIR

This project is the official implementation of 'Diffir: Efficient diffusion model for image restoration', ICCV2023
Apache License 2.0
451 stars 18 forks source link

The motion deblurring project seems to have a problem with the weights of the pre-trained model. #61

Closed DYY512 closed 2 months ago

DYY512 commented 2 months ago

RuntimeError: Error(s) in loading state_dict for DiffIRS1: Missing key(s) in state_dict: "E.E.0.weight", "E.E.0.bias", "E.E.2.body.0.weight", "E.E.2.body.0.bias", "E.E.2.body.2.weight", "E.E.2.body.2.bias", "E.E.3.body.0.weight", "E.E.3.body.0.bias", "E.E.3.body.2.weight", "E.E.3.body.2.bias", "E.E.4.body.0.weight", "E.E.4.body.0.bias", "E.E.4.body.2.weight", "E.E.4.body.2.bias", "E.E.5.body.0.weight", "E.E.5.body.0.bias", "E.E.5.body.2.weight", "E.E.5.body.2.bias", "E.E.6.body.0.weight", "E.E.6.body.0.bias", "E.E.6.body.2.weight", "E.E.6.body.2.bias", "E.E.7.weight", "E.E.7.bias", "E.E.9.weight", "E.E.9.bias", "E.E.11.weight", "E.E.11.bias", "E.mlp.0.weight", "E.mlp.0.bias", "E.mlp.2.weight", "E.mlp.2.bias". Unexpected key(s) in state_dict: "condition.E.0.weight", "condition.E.0.bias", "condition.E.2.body.0.weight", "condition.E.2.body.0.bias", "condition.E.2.body.2.weight", "condition.E.2.body.2.bias", "condition.E.3.body.0.weight", "condition.E.3.body.0.bias", "condition.E.3.body.2.weight", "condition.E.3.body.2.bias", "condition.E.4.body.0.weight", "condition.E.4.body.0.bias", "condition.E.4.body.2.weight", "condition.E.4.body.2.bias", "condition.E.5.body.0.weight", "condition.E.5.body.0.bias", "condition.E.5.body.2.weight", "condition.E.5.body.2.bias", "condition.E.6.body.0.weight", "condition.E.6.body.0.bias", "condition.E.6.body.2.weight", "condition.E.6.body.2.bias", "condition.E.7.weight", "condition.E.7.bias", "condition.E.9.weight", "condition.E.9.bias", "condition.E.11.weight", "condition.E.11.bias", "condition.mlp.0.weight", "condition.mlp.0.bias", "condition.mlp.2.weight", "condition.mlp.2.bias", "denoise.resmlp.0.weight", "denoise.resmlp.0.bias", "denoise.resmlp.2.resmlp.0.weight", "denoise.resmlp.2.resmlp.0.bias", "diffusion.betas", "diffusion.alphas_cumprod", "diffusion.alphas_cumprod_prev", "diffusion.sqrt_alphas_cumprod", "diffusion.sqrt_one_minus_alphas_cumprod", "diffusion.log_one_minus_alphas_cumprod", "diffusion.sqrt_recip_alphas_cumprod", "diffusion.sqrt_recipm1_alphas_cumprod", "diffusion.posterior_variance", "diffusion.posterior_log_variance_clipped", "diffusion.posterior_mean_coef1", "diffusion.posterior_mean_coef2", "diffusion.model.resmlp.0.weight", "diffusion.model.resmlp.0.bias", "diffusion.model.resmlp.2.resmlp.0.weight", "diffusion.model.resmlp.2.resmlp.0.bias", "diffusion.condition.E.0.weight", "diffusion.condition.E.0.bias", "diffusion.condition.E.2.body.0.weight", "diffusion.condition.E.2.body.0.bias", "diffusion.condition.E.2.body.2.weight", "diffusion.condition.E.2.body.2.bias", "diffusion.condition.E.3.body.0.weight", "diffusion.condition.E.3.body.0.bias", "diffusion.condition.E.3.body.2.weight", "diffusion.condition.E.3.body.2.bias", "diffusion.condition.E.4.body.0.weight", "diffusion.condition.E.4.body.0.bias", "diffusion.condition.E.4.body.2.weight", "diffusion.condition.E.4.body.2.bias", "diffusion.condition.E.5.body.0.weight", "diffusion.condition.E.5.body.0.bias", "diffusion.condition.E.5.body.2.weight", "diffusion.condition.E.5.body.2.bias", "diffusion.condition.E.6.body.0.weight", "diffusion.condition.E.6.body.0.bias", "diffusion.condition.E.6.body.2.weight", "diffusion.condition.E.6.body.2.bias", "diffusion.condition.E.7.weight", "diffusion.condition.E.7.bias", "diffusion.condition.E.9.weight", "diffusion.condition.E.9.bias", "diffusion.condition.E.11.weight", "diffusion.condition.E.11.bias", "diffusion.condition.mlp.0.weight", "diffusion.condition.mlp.0.bias", "diffusion.condition.mlp.2.weight", "diffusion.condition.mlp.2.bias".