Fixed error from calling common_metrics in validation step

openclimatefix · Dec 13, 2023 · 43b0285 · 43b0285
1 parent adeb778
commit 43b0285
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py
@@ -370,7 +370,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, targets=y)
+        common_metrics_each_step = common_metrics(predictions=y_hat.numpy(), target=y.numpy())
         mse_each_step = common_metrics_each_step["rmse"] ** 2
         mae_each_step = common_metrics_each_step["mae"]