论文

代码

论文原作者的代码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

EagleEye论文+代码相关推荐

进阶必备：CNN经典论文代码复现 | 附下载链接
经常会看到类似的广告<面试算法岗,你被要求复现论文了吗?>不好意思,我真的被问过这个问题.当然也不是所有面试官都会问,究其原因,其实也很好理解.企业肯定是希望自己的产品是有竞争力,有卖点的 ...
AI论文代码强制开源？“我反对！”
晓查发自凹非寺量子位报道 | 公众号 QbitAI AI论文,到底该不该强制开源代码? NeurIPS 2020官方近日发布了一份开源指南.从2019开始,NeurIPS就"强烈建 ...
最新+经典+论文+代码+多版本+神器+https://paperswithcode.com/
最新+经典+论文+代码+多版本+神器+https://paperswithcode.com/ Papers with Code 中收集了各种机器学习的内容:论文,代码,结果,方便发现和比较.通过这些数 ...
Chrome 科研神器！论文代码快速复现
点击上方,选择星标或置顶,不定期资源大放送! 阅读大概需要5分钟 Follow小博主,每天更新前沿干货导读:今天给大家推荐一款论文代码Chrome神器,这款插件叫做Find Code for Res ...
高清变脸更快更逼真！比GAN更具潜力的可逆生成模型来了 | OpenAI论文+代码
昨天上市即破发的小米,今天上午股价大涨近10%.这下雷军要笑了. 而且可以笑得更灿烂.更灿烂是什么样?来,我们用OpenAI刚刚发布的人工智能技术,给大家展示一下. 当然这个最新的技术,不止这 ...
2021年三大顶会时间序列论文代码整理
作者:杰少,炼丹笔记嘉宾 2021年最新时间序列预测论文&代码整理 AAAI 2021 Deep Switching Auto-Regressive Factorization: Applic ...
CVPR 2020 Oral 汇总：论文 / 代码 / 解读（更新中）
在2月24日,CVPR 2020 公布接收论文结果公布,从 6656 篇有效投稿中录取了 1470 篇论文,录取率约为 22%.3月13日,CVPR Oral结果公布了.有大佬已经分享了自己的工作,本 ...
撒花！算法岗必须复现的上百篇经典论文代码完结啦！ | 附下载链接
经常会看到类似的广告<面试算法岗,你被要求复现论文了吗?>不好意思,我真的被问过这个问题.当然也不是所有面试官都会问,究其原因,其实也很好理解.企业肯定是希望自己的产品是有竞争力,有卖点的 ...
【算法竞赛学习】学术前沿趋势-论文代码统计
任务3:论文代码统计 3.1 任务说明任务主题:论文代码统计,统计所有论文出现代码的相关统计: 任务内容:使用正则表达式统计代码连接.页数和图表数据: 任务成果:学习正则表达式统计: 3.2 数据处 ...
牛人主页（主页有很多论文代码）【真的好强大】
转自:http://blog.sina.com.cn/s/blog_6833a4df01012bcf.html 牛人主页(主页有很多论文代码) Serge Belongie at UC San Die ...

EagleEye论文+代码