-
Notifications
You must be signed in to change notification settings - Fork 10
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
cifar10 example error #4
Comments
Hi @qiuyang163, thanks for reaching out. The script works on my side.
My env: pytorch==1.13.1, torch_pruning==1.3.1. BTW, the |
yes,I got it,I succed in torch==1.13.1. But if I update to torch2.0, it failed |
Thank you, will fix it soon. |
run
bash scripts/prune_ddpm_cifar10.sh 0.3
then I got error:
I seems that q、k、v in_features is not equal to out_features
@VainF
The text was updated successfully, but these errors were encountered: