SSKD:[ECCV2020]知识提炼遇到自我监督 源码

普通话 此回购协议是论文《 (ECCV 2020)的实现。 先决条件 此存储库已在Ubuntu 16.04.5,Python 3.7,PyTorch 1.5.0,CUDA 10.2中进行了测试。使用此存储库之前,请确保安装pytorch,torchvision,tensorboardX,numpy。 跑步 教师培训 教师培训的一个例子是: python teacher.py --arch wrn_40_2 --lr 0.05 --gpu-id 0 您可以在其中通过标志--arch指定体系结构 您也可以下载所有经过预训练的教师模型。如果要直接运行student.py ,则必须重新组织目录。例如,当你下载vgg13.pth,你必须做出一个目录,再说teacher_vgg13,然后进行内部teacher_vgg13一个新的目录CKPT。将vgg13.pth移到Teacher_vgg13 / c