EagleEye论文代码论文代码数据集执行search.py文件执行choose_strategy.py文件执行finetune.py文件执行inference.py文件

论文代码

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

数据集ILSVRC2012数据集

下载ILSVRC2012数据集。 您必须先注册帐户并登录。 也可以直接替换为ImageNet数据集。

下载的ILSVRC2012数据集是三个. tar后缀的软件包文件，需要解压缩。

tar -xvf *.tar解压缩后的所有test和val都是直接的. jpeg图像，但train文件夹下仍然是. tar包。 因为我不知道如何递归地解包，所以我自己手写了python文件来解包。 打开包后，train文件夹中的所有内容都是. jpeg图像。

但是，在执行代码时会发生错误。 主要是文件夹层次结构错误。 这里列出了参考链接1、链接2两个。

如果你认为官方网站下载速度太慢，可以使用迅雷链接或百度云链接。 网上有很多。 这里推荐的是Linux多线程命令行下载工具aria2。 稳定在5MB/s秒。tiny-ImageNet数据集

tinyimagenetvisualrecognitionchallenge

运行search.py文件的问题1

运行时错误器3360 cuda运行时错误器(11 ) : invalidargumentat/py torch/aten/src/thc/thc general.CPP 3360383

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

运行choose_strategy.py文件时，将对上一步骤中获得的pruning_strategy.txt文件进行排序，并输出分数最高的前五个strategy。

finetune.py文件的运行问题1

运行时错误： cudnnerror : Cu dnn _ status _ execution _ failed

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

torch.backends.Cu dnn.enabled=false执行inference.py文件问题1

导出git _ python _ refresh=quiet

这似乎是git相关的问题。

解决方案

康达安装千兆问题2

symbollookuperror 3360 XXX/lib/python 3.6/site-packages/torch/lib/lib torch _ python.so 3360 undefined symbol 3360

python版本的理由从python3.6.0更新为python3.6.2即可。

解决方案

康达安装python=3.6. 2问题3

运行时错误器3360 cuda运行时错误器(11 ) : invalidargumentat/py torch/aten/src/thc/thc general.CPP 3360405

错误的原因可能是多GPU测试，也可能是PyTorch版本与显卡不兼容。

解决方案

将torch.backends.Cu dnn.benchmark=true (此语句一般出现在主函数的前几个语句中)更改为torch.backends.Cu dnn.benchmark=false。

参考

[ solved ] export git _ python _ refresh=quiet

symbollookuperror:undefined symbol:py slice _ unpack

运行时错误器3360 cuda运行时错误器(11 ) : invalidargumentat/py torch/aten/src/thc/thc general.CPP 3360383

如何使用. whl文件安装torch和pytorch

Ubuntu解决e :启用到位置包

Pytorch的torch vision.datasets.image folder中的Found 0 files in subfolders错误

导入解决pytorch使用问题的数据集. imagefolderimagenet数据集时出错

cuda error 3360 runtime error : Cu dnn _ status _ execution _ failed