5.4 计算训练损失