在ssd.pytorch网咯中训练自制的数据集,只训练一类,在voc0712.py脚本下更改数据集类别,图片中注释的部分是原来VOC数据集部分的代码,将里面的类别改成自己的后,因为我只训练一类,所以在括号外面再加一个中括号,解决untimeError: copy_if failed to synchronize: device-side assert trig报错的问题。
在ssd.pytorch网咯中训练自制的数据集,只训练一类,在voc0712.py脚本下更改数据集类别,图片中注释的部分是原来VOC数据集部分的代码,将里面的类别改成自己的后,因为我只训练一类,所以在括号外面再加一个中括号,解决untimeError: copy_if failed to synchronize: device-side assert trig报错的问题。