classdit = {‘airplane’: 0, ‘airport’: 1, ‘baseball_diamond’: 2, ‘basketball_court’: 3, ‘beach’: 4, ‘bridge’:5,‘chaparral’:6
losses.update(loss.item(), inputs.size(0))计算loss时报错
标签需要从0开始,按照顺序
RuntimeError: CUDA error: device-side assert triggered解决办法
最新推荐文章于 2023-06-25 02:15:25 发布