search

gmh14 / RobNets

[CVPR 2020] When NAS Meets Robustness: In Search of Robust Architectures against Adversarial Attacks
http://www.mit.edu/~yuzhe/robnets.html
MIT License
123 stars 15 forks source link

Could you open source the whole training code? #1

Closed HLinChen closed 4 years ago

HLinChen commented 4 years ago

Could you please open source the whole training process? I think that would more valuable than this fixed model. Thank you for your time and consideration.

gmh14 commented 4 years ago

Hi! Sorry for the late reply. Since we use a private distributed training framework for RobNets, it costs some time to convert them to Pytorch official distributed training and align the results. For now, the training scripts for CIFAR10 have been pushed. We are working on ImageNet, but mostly the code is based on https://github.com/ashafahi/free_adv_train. We will further release the model checkpoint on the clean dataset since it is important to the adversarial training on ImageNet. Thanks for your patience and consideration!

HLinChen commented 4 years ago

Thanks for your reply! I will try it.