用nvidia-docker跑深度学习模型

背景 最近实验室要参加一个目标检测的比赛,这段时间一直在跑ssd模型,最开始根据作者给的文档成功编译后,可以在VOC数据集上进行训练。由于要用比赛官方的数据集,因此做了几天的数据集,然后拿自己的数据集训练的时候,出现了以下报错:Check failed: a <= b (0 vs. -1.192093-07) 去网上搜了相关的解决方法,全都是说把math_functions.cpp第250行注释掉
相关文章
相关标签/搜索