MingSun-Tse / TPP

[ICLR'23] Trainability Preserving Neural Pruning (PyTorch)
31 stars 2 forks source link

hyperparameter of Table2 #2

Closed JingyangXiang closed 1 year ago

JingyangXiang commented 1 year ago

Thank you very much for your excellent work!I want to reproduce the results of Tab2. Could you please share your hyperparameters? Thank you very much!

MingSun-Tse commented 1 year ago

Hi @JingyangXiang thanks for your interest! The scripts to reproduce are already in README: https://github.com/MingSun-Tse/TPP#tab-2-resnets-on-imagenet

Please just follow the README exactly. Let me know if you have further questions!