自由Man

Deep Learning (pyTorch) 报错笔记

【2019.11.13-1】

  问题:RuntimeError: CUDA ERROR: an illegal memory access was encountered

  定位解决:ground truth中有类别超了,计算loss时需要添加ignore_index来忽略对应类别。

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

Powered By Z-BlogPHP 1.5.2 Zero Theme By 爱墙纸

Copyright ZiYouMan.cn. All Rights Reserved. 蜀ICP备15004526号