Yolo_踩的坑
Config:
数据集处理:
- OS:Mac OS X 10.14
跑模型:
- OS:Ubuntu 16.04
- GPU:GetForce GTX 960M
- NVIDIA Driver:430.09
- Cuda:10.1
- Cudnn:7.5.1
安装并没有太多问题,准备跑自己的模型的时候还是遇到了蛮多问题
推一下官方教程:https://pjreddie.com/darknet/yolo/
1. cuda error: out of memory darknet: ./src/cuda.c:36: check_error: Assertion `0’ failed.
改一下cfg,减小batch,增大subdivisions