# 损失函数 交叉熵
loss = nn.CrossEntropyLoss(reduction='none')
使用上述代码时,图像无法绘制(值设为sum
也不可以),将值改为mean
之后可以正常跑通
# 损失函数 交叉熵
loss = nn.CrossEntropyLoss(reduction='mean')
# 损失函数 交叉熵
loss = nn.CrossEntropyLoss(reduction='none')
使用上述代码时,图像无法绘制(值设为sum
也不可以),将值改为mean
之后可以正常跑通
# 损失函数 交叉熵
loss = nn.CrossEntropyLoss(reduction='mean')