ssd+pytorch1.2+visdrone

用的https://github.com/amdegroot/ssd.pytorch这一版本。 复现过程遇到的问题： 1、遇到CUDA error: out of memory      gpu内存不够，用 watch -n 2 nvidia-smi 监控一下gpu使用情况，2表示的是2秒刷新一下，然后指定一下空着的卡， 用这个以下命令，在train.py中指定一下空着的文件。 import os

>>阅读原文<<