Giters

arpitbansal297 / Cold-Diffusion-Models

Official implementation of Cold-Diffusion for different transformations in pytorch.

Home Page:https://arxiv.org/abs/2208.09392

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Kernel size mismatch

lujain197 opened this issue · comments

commented

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....

commented

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

commented

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?