- 博客(4)
- 资源 (2)
- 收藏
- 关注
原创 雷达视觉融合算法RODNet数据处理代码解读
雷视融合,使用RGB图像对雷达数据进行标注,然后使用标注数据监督学习,本文章主要讲解RODNet的数据处理部分
2023-07-27 15:17:32
3474
4
原创 pytorch在训练时,显示内存不够 cuda out of memory
在跑代码时,训练集跑的时候正常,在测试验证集和保存模型的时候,提示内存不够,直接报错终止,这种一般并不是batchsize过大导致的,不然训练集也运行不起来,最简单直接的方法,就是把验证集放到cpu上,一般验证集的数据比较小,也很快就完成了例如:imgn_val =Variable(imgn_val.cuda(), requires_grad=False)out_val = torch.clamp(model(imgn_val), 0., 1.)这里把验证集转到cpu里,去掉 .cuda()即可
2021-11-28 19:17:10
3052
2
原创 李宏毅2020深度学习HomeWork1数据集
李宏毅2020深度学习HomeWork1数据集数据附下链接:https://pan.baidu.com/s/181NSI03okxfl4vu2_Ep6-g提取码:fl1a复制这段内容后打开百度网盘手机App,操作更方便哦
2020-12-14 10:40:57
603
yolov5 摔倒检测 跌倒识别
2022-07-13
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人