Requirements:

Python ≥ 3.6
PyTorch ≥ 1.4
pip install tensorboard_logger torchnet faiss-gpu

Hyper-parameters and model configurations are located in ./config

In order to run experiments, please download the corresponding dataset and place it at the location specified in the config file. Then run python main.py --exp [config_file].

For example, run the following command to reproduce the paper's result on CIFAR-10:

1. 50% symmetric noise: python main.py --exp cifar10_sym
2. 40% asymmetric noise: python main.py --exp cifar10_asym


Thank you!

