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