Remove debugging

openclimatefix · Jan 24, 2024 · 0c6b9c0 · 0c6b9c0
1 parent 595c1df
commit 0c6b9c0
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py
@@ -464,14 +464,8 @@ def validation_step(self, batch: dict, batch_idx):
         """Run validation step"""
         # Make all -1 values 0.0
         batch[self._target_key] = batch[self._target_key].clamp(min=0.0)
-        print(f"batch shape: {batch[self._target_key].shape}")
-        print(f"Shape after slice: {batch[self._target_key][:, -self.forecast_len_30 :, 0].shape}")
         y_hat = self(batch)
         # Sensor seems to be in batch, station, time order
-        print(f"y_hat shape: {y_hat.shape}")
-        print(f"batch shape: {batch[self._target_key].shape}")
-        print(f"Shape after slice: {batch[self._target_key][:, -self.forecast_len_30 :, 0].shape}")
-        exit()
         y = batch[self._target_key][:, -self.forecast_len_30 :, 0]
 
         losses = self._calculate_common_losses(y, y_hat)