在ssd.pytorch网咯中训练自制的数据集,只训练一类,在voc0712.py脚本下更改数据集类别,图片中注释的部分是原来VOC数据集部分的代码,将里面的类别改成自己的后,因为我只训练一类,所以在括号外面再加一个中括号,解决untimeError: copy_if failed to synchronize: device-side assert trig报错的问题。
001_在ssd.pytorch网络上训练自己的数据集,出现报错RuntimeError: copy_if failed to synchronize: device-side assert trig
最新推荐文章于 2024-05-13 10:42:06 发布