No.62584427 - 技术宅
无标题无名氏No.62584427 只看PO
2024-05-31(五)09:45:10
ID:KcuTIDY
用torch训练，在计算损失的时候，如果预测值和实际值是多维的，比如（16，3），那torch的criterion是怎么计算损失的，挨个算然后求均值吗( ﾟ∀。)？
- 上一页 [
- 1 ]
- 下一页
无标题无名氏No.62584427 只看PO
2024-05-31(五)09:45:10
ID:KcuTIDY 回应
用torch训练，在计算损失的时候，如果预测值和实际值是多维的，比如（16，3），那torch的criterion是怎么计算损失的，挨个算然后求均值吗( ﾟ∀。)？
无标题无名氏No.62590378
2024-05-31(五)20:17:30 ID: C5A03Bf
>>No.62587777
不需要写处理输出数据的循环 torch提供的loss算法会自动做矩阵运算的 并且一般都是会自动帮你把batch的维度也算了
无标题无名氏No.62681171
2024-06-08(六)13:22:20 ID: MNcDEVl
用torch.nn.MSELoss就好
torch会帮你打理好所有问题，只管调用就好了ᕕ( ᐛ )ᕗ