回应模式 - No.62584427


No.62584427 - 技术宅


无标题无名氏No.62584427 只看PO

2024-05-31(五)09:45:10 ID:KcuTIDY 回应

用torch训练,在计算损失的时候,如果预测值和实际值是多维的,比如(16,3),那torch的criterion是怎么计算损失的,挨个算然后求均值吗( ゚∀。)?

Tips无名氏No.9999999

2099-01-01 00:00:01 ID: Tips

♡性♡感♡红♡名♡在♡线♡要♡饭♡
(〃∀〃) https://afdian.com/a/nmbxd

无标题无名氏No.62584491

2024-05-31(五)09:51:51 ID: C5A03Bf

是的 并且还要看你用的loss算法

无标题无名氏No.62587777

2024-05-31(五)15:58:03 ID: KcuTIDY (PO主)

损失函数就用的mse,是直接算均值的就行,我还想的手写个循环来着(; ゚ 3゚)

无标题无名氏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会帮你打理好所有问题,只管调用就好了ᕕ( ᐛ )ᕗ