search

Peter-Kocsis / IntrinsicImageDiffusion

Intrinsic Image Diffusion for Single-view Material Estimation
Other
166 stars 13 forks source link

Error in the released checkpoint for material diffusion #10

Open HollyDQWang opened 1 month ago

HollyDQWang commented 1 month ago

Hi! Congrats on the great work and thanks for the code and model ckpt. When testing inference on material decomposition, I encountered the following issue: Error executing job with overrides: ['logger=console'] Traceback (most recent call last): File "/home/IntrinsicImageDiffusion/iid/material_diffusion/__main__.py", line 141, in main material_diffusion(cfg) File "/home/IntrinsicImageDiffusion/iid/material_diffusion/__main__.py", line 70, in material_diffusion model.load_state_dict(ckpt) File "/home/Documents/IntrinsicImageDiffusion/iid/material_diffusion/iid.py", line 315, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for IntrinsicImageDiffusion: Missing key(s) in state_dict: "diffusion_module.logvar".

I thus wonder whether there's issue with the released model checkpoints?

HollyDQWang commented 1 month ago

I set strict=False to fix this error but still not sure why it happens in the first place.

fantasy-byakuya commented 1 month ago

I have same question