VainF/Diff-Pruning

ddpm_prune.py errored out at step 249

vuiseng9 opened this issue · 2 comments

vuiseng9 commented

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 commented

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

Step 1: Pruning:
image

Step 2: Finetuning
image

VainF commented

BTW, my environment:

pytorch==1.12.1
torchvision==0.13.1