wang959512421的博客

本文主要针对yolo_v3进行模型剪枝量化，主要使用voc格式的数据集训练自己的数据，代码主要在：-[x]https://github.com/ReOneK/Yolo_v3_prune_vocdata具体步骤如下：## 一、数据处理- 将标注数据放入data/Annotations,图片数据放入data/images- 运行modify.py文件，避免因图片问题导致标注文件中图片的长宽为0的问题-将voc_label.py中classes修改为自己的训练类别，并分别运行make..

最近无聊发现李沐大神的深度学习资料，于是也想着手学习一下mxnet,本以为只是 在anaconda下简单的pip install mxnet_cu90,但是发现不能成功调用GPU,弄了大半天也不行，最终找到如下可用办法： 1   git clone https://github.com/mli/gluon-tutorials-zh2  conda create -n gluon a...

原本安装pytorch只需要一条命令：conda install pytorch torchvision cudatoolkit=9.0 -c pytorch结果不是网不行就是各种出错。。。于是用pip安装试了一下，发现还是pip好用啊，conda说的那么好。。。这是官网推荐的pip install https://download.pytorch.org/whl/c...

关于torch.gather函数直接上代码example1输出为example2输出为

首先①使用 torch.cuda.is_available()查看是否正确安装cuda以及cudnn,但是有时候会发现明明是有GPU存在就是调用不了，出现以下错误PyTorch非常容易的就可以使用GPU，可以用如下方式把一个模型放到GPU上：device = torch.device("cuda：0")model.to(device)然后可以复...

看了无数篇博客都是牛头不对马嘴，忙活了半天才搞清楚如果大家在用torch的过程中看见这样的错误。。。。淡定，不是你的问题，也不是代码问题。。。是破win10的原因只需要在代码前面加上torch.cuda.current_device()torch.cuda._initialized = True就可以正常训练了。。。心累啊...