PyTorch螺旋数据集二分类网络Loss不下降、准确率0%问题排查
PyTorch二分类逻辑回归训练异常排查
我在PyTorch中实现了一个二分类逻辑回归网络,采用YouTube博主sentdex《从零开始构建神经网络》中的螺旋数据集训练,但Loss几乎无下降,准确率始终为0%。我的网络结构与博主的手写实现几乎一致,而他的模型能正常训练,因此我怀疑是PyTorch代码中的Bug导致问题。
现附上以下内容,请求协助排查:
- 我的PyTorch实现代码
- 我的训练输出日志
- 博主的训练输出日志
内容的提问来源于stack exchange,提问作者Tristan Pagden
相关产品推荐
相关产品推荐

