Files
xautodl/xautodl/procedures/simple_KD_main.py