Robust-NN-Training

Implementation for the proposed formulation of robust neural network training, in Advances in Neural Information Processing Systems 32 (NIPS 2019) Link to Paper

Requirements

• Python >= 3.5 or Python 2.7
• PyTorch >= 1.3 (to use pretrained_model) or PyTorch >= 1.0 (to train from scratch)

Getting Started

python3 main.py