EagleEye论文+代码

2022-09-22 20:49:38 浏览数 (1)

大家好，又见面了，我是你们的朋友全栈君。

EagleEye论文 代码

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

论文

代码

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

数据集

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

代码语言：javascript复制
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版本，失败。最后选择了添加一句语句，成功。

代码语言：javascript复制
torch.backends.cudnn.enabled=False

执行inference.py文件

问题一

export GIT_PYTHON_REFRESH=quiet

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

代码语言：javascript复制
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即可。 解决方案

代码语言：javascript复制
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.cn/170731.html原文链接：https://javaforall.cn

python pytorch 深度学习 c++ c#

0 人点赞