![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
yolo系列训练数据集
文章平均质量分 75
oyall520
这个作者很懒,什么都没留下…
展开
-
如何将模型训练中断生成的多个tf.events文件接到一起
最近跑深度学习模型,就很邪门一直中断,跑到一半就停了然后报错 “cuda out of memory”,通过百度综合了很多大神的解决办法。1. 处理 train.py 文件在运行模型的地方,加上 torch.cuda.empty_cache() 删除一些不需要的变量。原代码:# Run modelpred = model(imgs)修改之后:# Run modeltry: pred = model(imgs)except RuntimeError as exception:原创 2021-03-11 11:41:09 · 1912 阅读 · 0 评论 -
pytorch-yolov4训练自己的数据集之环境配置
使用的项目地址:https://github.com/Tianxiaomo/pytorch-YOLOv4一、配置环境1.因为不同的项目所需要的环境也不同,每次都更新太费神,所以直接在anaconda下建一个虚拟环境,给环境起名叫 yolov4:conda create -n yolov4 python=3.7因为我的 anaconda 自带的 python 就是3.7版本,所以这里安装3.7。2.激活虚拟环境上面的命令执行后会出现如下提示:故我们输入conda activate yolov原创 2021-01-09 20:24:30 · 1581 阅读 · 9 评论