ddpm_prune.py errored out at step 249

[vuiseng9]

Hello there, thank you for the interesting work!

I was following the readme to learn about your method. While running step 1, it errored out at this line with message RuntimeError: shape '[1, -1, 1, 192]' is invalid for input of size 45824. It only happened at step 249/1000, prior steps were fine. Wonder if you face this or have some idea about this. Thank you!

[VainF]

Hi, @vuiseng9. The pruning and finetuning work well on my side.

Step 1: Pruning:

Step 2: Finetuning

[VainF]

BTW, my environment:

pytorch==1.12.1
torchvision==0.13.1

[jhangyeol]

Still getting same issue. In my case,

RuntimeError: shape '[1, -1, 1, 160]' is invalid for input of size 45824 at step 256/1000.

My environment is:

torch 2.7.1
torch-pruning 1.6.0
torchvision 0.22.1