arpitbansal297 / Cold-Diffusion-Models

Official implementation of Cold-Diffusion for different transformations in pytorch.
https://arxiv.org/abs/2208.09392
1k stars 79 forks source link

Kernel size mismatch #12

Open lujain197 opened 1 year ago

lujain197 commented 1 year ago

I''ve been trying testing blurring using weights on the drive,

! python ./AFHQ_128_test.py --load_path ./AFHQ_blur_generation.pt --gmm_cluster 1 --noise 0.002 --discrete --time_steps 10 --blur_size 15 --blur_std 0.01 --blur_routine Individual_Incremental --sampling_routine x0_step_down --data_path ./test_data --save_folder ./save_folder --test_type test_data

but I always get this error

size mismatch for module.gaussian_kernels.0.weight: copying a param with shape torch.Size([3, 1, 27, 27]) from checkpoint, the shape in current model is torch.Size([3, 1, 1, 1]) size mismatch for module.gaussian_kernels.1.weight: copying a param with shape torch.Size([3, 1, 27, 27]) from checkpoint, the shape in current model is torch.Size([3, 1, 3, 3]). size mismatch for module.gaussian_kernels.2.weight: copying a param with shape torch.Size([3, 1, 27, 27]) from checkpoint, the shape in current model is torch.Size([3, 1, 5, 5]). size mismatch for module.gaussian_kernels.3.weight: copying a param with shape torch.Size([3, 1, 27, 27]) from checkpoint, the shape in current model is torch.Size([3, 1, 7, 7]). size mismatch for module.gaussian_kernels.4.weight: copying a param with shape torch.Size([3, 1, 27, 27]) from checkpoint, the shape in current model is torch.Size([3, 1, 9, 9]). size mismatch for module.gaussian_kernels.5.weight: copying a param with shape torch.Size([3, 1, 27, 27]) from checkpoint, the shape in current model is torch.Size([3, 1, 11, 11]). size mismatch for module.gaussian_kernels.6.weight: copying a param with shape torch.Size([3, 1, 27, 27]) from checkpoint, the shape in current model is torch.Size([3, 1, 13, 13]). size mismatch for module.gaussian_kernels.7.weight: copying a param with shape torch.Size([3, 1, 27, 27]) from checkpoint, the shape in current model is torch.Size([3, 1, 15, 15]). size mismatch for module.gaussian_kernels.8.weight: copying a param with shape torch.Size([3, 1, 27, 27]) from checkpoint, the shape in current model is torch.Size([3, 1, 17, 17]). size mismatch for module.gaussian_kern....

arpitbansal297 commented 1 year ago

Hi, This is because of the blur_routine. Can you see if the problem persists if you use "--blur_routine Exponential"?

lujain197 commented 1 year ago

yes I still getting the error could you please help me to test the blurring weights on my data and I got error on this library gmm_pycave that it is not found , how can I get it?