pytorch走过的坑 发表于 2019-10-06 | 分类于 pytorch | 字数统计 52 | 阅读时长 1 记录一下学习pytorch遇到的两大坑 zero_grad 需要收到清空历史梯度 torch.nn.CrossEntropyLoss 不用加入激活层softmax，在运算时已经加入了，如果加入softmax用nnl损失