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

>>阅读原文<<