大家好，又见面了，我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。
Jetbrains全系列IDE使用 1年只要46元 售后保障 童叟无欺
EagleEye论文+代码
论文
代码
论文原作者的代码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
发布者：全栈程序员-站长，转载请注明出处：https://javaforall.net/170731.html原文链接：https://javaforall.net
