search

sjtuplayer / few-shot-diffusion

[ICCV 2023] Phasic Content Fusing Diffusion Model with Directional Distribution Consistency for Few-Shot Model Adaption
49 stars 2 forks source link

fusion model &ckpt #12

Open bigwahaha opened 3 months ago

bigwahaha commented 3 months ago

感谢您的工作！请问这两步中训练使用的数据集都是source data吗？ image

另外，我在加载您提供的训练好的checkpoint时，出现参数不匹配的情况。请问您使用的UNet网络是您现在代码所提供的吗？我不想重新进行训练，能否麻烦您帮助我解决一下这个问题？ image

sjtuplayer commented 2 months ago

你可以试试重装一下denoising_diffusion_pytorch==0.28.0