henryzhongsc / adv_robust_gkp Goto Github PK

Hi @henryzhongsc, thanks for your work on this repo.

I was wondering whether it could be possible to train a different model than a ResNet20 with your current state of the code.
I am trying to prune a ResNet18, but looks like I cannot perform the standard pretraining with this current code state:
2024-02-19 15:39:53,474 | ERROR : Input task <train> is not supported.
Likewise, it is a bit hard to load the state dict of an externally pretrained model into your ModifiedResNet class in prune_cifar.py (line 33)

Any chance you are planning to solve the problems in the code, or load some pretrained?
Thanks,
Giorgio