踩坑
文章平均质量分 58
qq_43222384
这个作者很懒,什么都没留下…
展开
-
LSTR 效果复现踩坑
论文链接:https://arxiv.org/pdf/2011.04233.pdf论文出处:WACV2020论文代码:https://github.com/liuruijin17/LSTR研发团队:西安交通大学人工智能与机器人研究所过程踩坑:基于tusimple 数据集从零训练训练,过程中出现 loss 下降至一个局部最优解,此时loss 在15 附近,无法继续收敛下降 。原因分析: 使用的Adam 优化器,batch_size=8, lr 从1 ,开始,想法是开始用大学习率让loss 快速下原创 2021-11-11 16:43:50 · 1241 阅读 · 3 评论 -
Corrupt JPEG data: premature end of data segment
在跑 LSTR 的效果时 , 用 tusimple数据集(1280720)正常,换成自集的数据集(12801080),出现 Corrupt JPEG data: premature end of data segment知乎回答:在所有图片进行检查后,觉得是第3 条的原因, 在传入网络前进行数据增广时 resize操作,并未去除, 不影响训练和效果。https://www.zhihu.com/question/303726551.遇到了同样的问题。针对jpg文件来说,一般是由于文件下载不完整导致原创 2021-11-11 15:20:55 · 3308 阅读 · 0 评论 -
pytorch模型转成ONNX
使用conda 虚拟环境Python3.6.8 Pytorch1.1版本, torchvision 0.3.0版本安装:pip install onnx onnxruntimeyolov3 pytorch 模型转onnx 代码:参考pytorch教程:https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html# Some standard importsfrom models import *im原创 2021-02-10 20:51:45 · 623 阅读 · 0 评论 -
报错处理整理
RuntimeError:_thnn_conv2d_forward not supported on CUPType for ByteIt seems you are trying to pass an input as a ByteTensor (uint8), which is not supported.Could you call input = input.float() before passing it to the modelRuntimeError:Input type(torch.原创 2020-07-02 10:37:10 · 367 阅读 · 0 评论 -
报错:no CUDA-capable device is detected 解决
一个星期没跑模型,今天开始跑模型时,在x =.cuda() 处报错用 if torch.cuda.is_available(): 判断 是 FLASE报错:no CUDA-capable device is detected控制台nvidia-smi报错:Failed to initialize NVML: Driver/library version mismatch解决办法:闭...原创 2019-08-08 10:45:02 · 3801 阅读 · 0 评论