The trained ResNet18 model on ImageNet is saved as ResNet18_ImageNet.pth.

ATKD is implemented in distiller_zoo/KD.py

Teacher training:
python3 train_teacher.py --model resnet56 --trial 1

Student training;
python3 train_student.py --path_t ./save/models/teacher_model_path --distill kd --model_s resnet20 -r 0 -a 1 -b 0 



