在使用pytorch训练的时候提示RuntimeError: copy_if failed to synchronize: device-side assert triggered错误
有两个方法可以尝试去解决一下:
1.尝试减少学习率试试看能不能解决这个问题,如果不能,请看第二种方法
2.看看config文件中设置的类别数目与数据中实际的类别数目是否一致,注意对于目标检测需要根据实际的类别数+1(背景)
本文分享自 作者个人站点/博客 前往查看
如有侵权,请联系 cloudcommunity@tencent.com 删除。
本文参与 腾讯云自媒体同步曝光计划 ,欢迎热爱写作的你一起参与!