深度学习
文章平均质量分 79
kongkongqixi
技术技术技术,非要科班嘛
展开
-
YOLOv2和YOLO9000论文总结
YOLOv2详解yolo模型和fast rcnn比有定位误差大,而且recall比较差,yolov2在关注准确率上提高定位和recall值对yolo进行改进,从几个方面进行改进:Batch Normalization:在所有卷积层后添加BN,可以加快了模型的收敛速度,取代了其他的正则化,还替代了dropout,同样可以防止过拟合,给模型带来了2%的mAP的改善; 高输出的分辨率:YOLO...原创 2019-03-31 23:49:28 · 800 阅读 · 0 评论 -
pytorch关于多块gpu使用总结,报错AssertionError: Invalid device id
pytorch默认使用gpu编号为device:0的设备,可以使用 torch.nn.DataParallel(model, device_ids=[0, 1]) 对模型进行制定gpu指定编号多gpu训练,必须要有编号为device:0的gpu,不然会报AssertionError: Invalid device id错误; 当gpu编号为device:0的设备被占用时,指定其他编号gpu...原创 2019-09-03 16:45:55 · 23846 阅读 · 14 评论