![](https://img-blog.csdnimg.cn/20201014180756916.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
DeepLearning
zhangjipinggom
俗人
展开
-
pytorch Dataloader 速度慢
1. 问题描述Dataloader每读完一个batch中间会间隔2s左右,很浪费时间,而且确定不是由于Dataset的__getitem__()复杂导致的。并且有一个规律,如果设置num_works=4,会出现4个batch一批很快,然后中间又间隔2s,再4个batch很快,再间隔。。。设成其他>0(多线程)的值同样规律。2. 解决办法等。。。其他CPU或者内存占用很大的程序结束。中间试过网上的prefetch,DataloaderX。基本没什么用。。。。3. 原因可能原因..原创 2020-06-30 11:29:23 · 6801 阅读 · 8 评论 -
cuda error
1. 问题描述想把从图像手工生成的特征和经过backbone提取的特征concatenate起来再进入网络后面的层:x_added_coord = torch.cat([x_backbone, x_coord], 1)报错:RuntimeError: CUDA error: an illegal memory access was encountered2. 解决办法:(1)只是用一个GPU(2)或者获知使用CPU训练...(不大现实)(3)或者把特征都放到cuda:0: x_...原创 2020-06-27 17:57:26 · 913 阅读 · 0 评论 -
【CVAT-安装-Windows】
后记: windows上没有安装成功,别往下看了。。。。。。。1. CVAT(Computer Vision Annotation Tool)简介免费的标注工具,可以标注视频,也可以标注图片https://github.com/opencv/cvat2. 安装过程:事实上cvat的官方网站https://github.com/opencv/cvat都有介绍,这里还是记录一下过程和...原创 2019-03-25 19:53:31 · 6649 阅读 · 2 评论