回应模式 - No.62584427


No.62584427 - 技术宅


无标题无名氏No.62584427 返回主串

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

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

无标题无名氏No.62587777

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

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