Skip to content
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

Reproduce the results #1

Open
smallwangzi opened this issue Sep 21, 2024 · 2 comments
Open

Reproduce the results #1

smallwangzi opened this issue Sep 21, 2024 · 2 comments

Comments

@smallwangzi
Copy link

I would greatly appreciate it if you could enlighten me on the difference between running commands with OMP and IMP, respectively. Thank you.

@shivank21
Copy link
Member

Thank you for your interest in our work
The main difference between OMP(One Shot Magnitude Pruning) and IMP(Iterative Magnitude Pruning) pruning running commands is that for OMP pruning, you need to prune the model a constant number of times(2). Whereas in IMP pruning, we run it for a larger number of time stamps. You can find the exact details here

@smallwangzi
Copy link
Author

Hello, I would like to ask a question regarding Class-wise forgetting. There is a significant difference between the results I obtained and those described in the paper. Can you help me out and tell me what the problem is?

python -u main_forget.py --seed=1 --gpu 0 --data ./datasets/cifar10 --dataset cifar10 --save_dir './_results/cifat10' --mask ./omp/cifar10/1model_SA_best.pth.tar --print_freq=10 --unlearn FT --class_to_replace 0 --unlearn_epochs 10 --unlearn_lr 0.01
FT
FF

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants