Debugging

openclimatefix · Dec 5, 2023 · 2d11c8b · 2d11c8b
1 parent 55e1957
commit 2d11c8b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py
@@ -449,6 +449,10 @@ def training_step(self, batch, batch_idx):
     def validation_step(self, batch: dict, batch_idx):
         """Run validation step"""
         y_hat = self(batch)
+        print(f"y_hat.shape: {y_hat.shape}")
+        print(f"{batch[self._target_key].shape}")
+        print(f"{batch[self._target_key][:, -self.forecast_len_30 :, 0].shape}")
+        print(f"{self.forecast_len_30}")
         y = batch[self._target_key][:, -self.forecast_len_30 :, 0]
 
         losses = self._calculate_common_losses(y, y_hat)