loss.backward()报错RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

最近在处理二分割问题时,遇到了标题所述的问题。

首先我可以确定我的环境绝对没问题。

仔细检查代码发现是 网络用了深度监督机制,即输出多级结果和gt一同算loss。

仔细检查代码发现,几级输出有两级忘记用sigmoid激活直接算BCE Loss所以导致报错。

你可能感兴趣的:(计算机幻觉,深度学习,pytorch,人工智能)