深度学习
qq_38775453
这个作者很懒,什么都没留下…
展开
-
2021-05-22
pytorch调试刚开始像正常一样用gpu调试,报错:RuntimeError: CUDA error: device-side assert triggered,只有前面有数组越界的错误,后面没了。choose_bg_inds =torch.tensor(np.random.choice((cls_inds_cpu).size()[0]), size=(choose_bg_num), replace=False),原创 2021-05-22 18:43:56 · 197 阅读 · 0 评论 -
深度学习训练loss为nan
深度学习训练loss为nan深度学习训练loss为nan深度学习训练loss为nan训练前几个epoch的损失都是正常的,后面出现很多nan。通过网上的经验,在loss函数后面加上tanh函数控制输出范围就可去除nan。原创 2021-05-14 08:50:04 · 274 阅读 · 0 评论