implementation of https://arxiv.org/abs/2104.03624
Install requirements
pip install -r requirements.txt
Train a model (CIFAR10)
python train.py \
--dataset CIFAR10 \
--network resnet34 \
--cuda \
--save_best \
--lr 0.01 \
--batch_size 64 \
--num_epochs 20
Transfer test for distorted CIFAR10
python transfer.py \
--dataset_to CIFAR10Distorted \
--network Unet \
--model_from models/CIFAR10_resnet34.ckpt \
--cuda \
--save_best \
--lr 0.01 \
--batch_size 64 \
--num_epochs 20