PR comment

Author: jacobbieker

Diff for: pvnet/models/base_model.py

@@ -390,7 +390,7 @@ def _calculate_val_losses(self, y, y_hat):
             # Take median value for remaining metric calculations
             y_hat = self._quantiles_to_prediction(y_hat)
         # common_metrics_each_step = common_metrics(predictions=y_hat.numpy(), target=y.numpy())
-        mse_each_step = torch.sqrt(torch.mean((y_hat - y) ** 2, dim=0)) ** 2
+        mse_each_step = torch.mean((y_hat - y) ** 2, dim=0)
         mae_each_step = torch.mean(torch.abs(y_hat - y), dim=0)
 
         losses.update({f"MSE_horizon/step_{i:03}": m for i, m in enumerate(mse_each_step)})