大家好，又见面了，我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。

Jetbrains全系列IDE使用 1年只要46元 售后保障 童叟无欺

论文

代码

论文原作者的代码GitHub链接为EagleEye。

数据集

ILSVRC2012数据集
下载ILSVRC2012数据集。需要先注册账号并登陆。也可以直接用ImageNet数据集代替。
下载好的ILSVRC2012数据集是3个.tar后缀的打包文件，需要解包。

tar -xvf *.tar

解压后的test和val都是直接的.jpeg图片，但是train文件夹下还是.tar包。我不知道怎么递归解包，所以自己手写了一个python文件解包。解包后，train文件夹中也全是.jpeg图片。
但是在执行代码的时候还是会报错，主要是文件夹的层次不对，这里给出两个参考链接链接一、链接二。
如果觉得从官网下载的速度太慢，可以使用迅雷链接或百度云链接，网上都有很多。在这里，我推荐使用一个Linux 多线程命令行下载工具aria2，可以达到稳定在5MB/s。
tiny-ImageNet数据集
Tiny ImageNet Visual Recognition Challenge

执行search.py文件

问题一

RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:383

可能是多GPU问题，或者pytorch与cuda版本不兼容。从whl文件重新安装torch和torchvision就好了。

执行choose_strategy.py文件

只是将上一步得到的pruning_strategy.txt文件排序，输出得分最高的前五个strategy。

执行finetune.py文件

问题一

RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

尝试过修改num_workers=0，失败；尝试过修改cuda版本，失败。最后选择了添加一句语句，成功。

torch.backends.cudnn.enabled=False

执行inference.py文件

问题一

export GIT_PYTHON_REFRESH=quiet

这好像是一个和git有关的问题。
解决方案

conda install git

问题二

symbol lookup error:xxx/lib/python3.6/site-packages/torch/lib/libtorch_python.so: undefined symbol: PySlice_Unpack

python版本原因，从python3.6.0更新到python3.6.2即可。
解决方案

conda install python=3.6.2

问题三

RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:405

错误原因可能是多GPU测试，或PyTorch版本与显卡不兼容。
解决方案

将 torch.backends.cudnn.benchmark = True(该句一般出现在主函数的开头几句)，改为 torch.backends.cudnn.benchmark = False。

参考
[solved] export GIT_PYTHON_REFRESH=quiet
symbol lookup error： undefined symbol：PySlice_Unpack

RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:383
使用.whl文件安装torch和pytorch的方法
Ubuntu 解决 E:Unable to locate package
Pytorch中torchvision.datasets.ImageFolder的Found 0 files in subfolders错误
解决pytorch使用问题数据集.ImageFolder导入Imagenet数据集时出错
Cuda Error : RuntimeError: CUDNN_STATUS_EXECUTION_FAILED