search

fangyuanmao / PID

Code for PID: Physics-Informed Diffusion Model for Infrared Image Generation
MIT License
53 stars 5 forks source link

About model checkpoint #5

Closed kay-kuo closed 1 month ago

kay-kuo commented 1 month ago

Hi, thanks for your great work.

When I load the checkpoint of KAIST, it reports Recovering from ./PID_KAIST/epoch=000235-step=000059999.ckpt with 299 missing and 1048 unexpected keys. Is this right? Thanks.

fangyuanmao commented 1 month ago

Thanks for your question. Please check if you have placed the VQVAE-f8 checkpoint and TeVNet checkpoint in right paths which are set in configs/latent-diffusion/kaist512-vqf8.yaml. There may be some unexpected keys in VQVAE-f8 pretrained model(it doesn't matter), but others have no anomalies.

kay-kuo commented 1 month ago

Thanks. Yeah, there is something wrong with the path of ckpts.