Merge pull request #176 from openclimatefix/horizon_graph_gpu_fix

dfulu · web-flow · commit d436b6482ef9 · 2024-04-15T10:51:06.000+01:00
Fix the horizon graph training on GPU
diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py
@@ -521,7 +521,7 @@ def validation_step(self, batch: dict, batch_idx):
 
         # Store these to make horizon accuracy plot
         self._horizon_maes.append(
-            {i: losses[f"MAE_horizon/step_{i:03}"] for i in range(self.forecast_len)}
+            {i: losses[f"MAE_horizon/step_{i:03}"].cpu().numpy() for i in range(self.forecast_len)}
         )
 
         logged_losses = {f"{k}/val": v for k, v in losses.items()}

Original file line number	Diff line number	Diff line change
`@@ -521,7 +521,7 @@ def validation_step(self, batch: dict, batch_idx):`
`521`	`521`
`522`	`522`	`# Store these to make horizon accuracy plot`
`523`	`523`	`self._horizon_maes.append(`
`524`		`- {i: losses[f"MAE_horizon/step_{i:03}"] for i in range(self.forecast_len)}`
	`524`	`+ {i: losses[f"MAE_horizon/step_{i:03}"].cpu().numpy() for i in range(self.forecast_len)}`
`525`	`525`	`)`
`526`	`526`
`527`	`527`	`logged_losses = {f"{k}/val": v for k, v in losses.items()}`