Fayeben / GenerativeDiffusionPrior

Generative Diffusion Prior for Unified Image Restoration and Enhancement (CVPR2023)
Apache License 2.0
277 stars 30 forks source link

RuntimeError: Error(s) in loading state_dict for UNetModel: #39

Open CharisWg opened 5 months ago

CharisWg commented 5 months ago

can you tell me how to solve this? UserWarning: Failed to load image Python extension: libc10_cuda.so: cannot open shared object file: No such file or directory warn(f"Failed to load image Python extension: {e}") os.path.expanduser(dir) is: /datashare3/charis/code2/Diffusion_models/GenerativeDiffusionPrior-main/output Logging to /datashare3/charis/code2/Diffusion_models/GenerativeDiffusionPrior-main/output creating model and diffusion... Traceback (most recent call last): File "sample_x0_blur.py", line 286, in main() File "sample_x0_blur.py", line 99, in main model.load_state_dict( File "/datashare3/charis/anaconda/envs/GDP2/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1482, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for UNetModel: Missing key(s) in state_dict: "input_blocks.3.0.op.weight", "input_blocks.3.0.op.bias", "input_blocks.4.0.skip_connection.weight", "input_blocks.4.0.skip_connection.bias", "input_blocks.6.0.op.weight", "input_blocks.6.0.op.bias", "input_blocks.7.1.norm.weight", "input_blocks.7.1.norm.bias", "input_blocks.7.1.qkv.weight", "input_blocks.7.1.qkv.bias", "input_blocks.7.1.proj_out.weight", "input_blocks.7.1.proj_out.bias", "input_blocks.8.1.norm.weight", "input_blocks.8.1.norm.bias", "input_blocks.8.1.qkv.weight", "input_bloc