Skip to content

Commit 0c6b9c0

Browse files
committed
Remove debugging
1 parent 595c1df commit 0c6b9c0

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

pvnet/models/base_model.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -464,14 +464,8 @@ def validation_step(self, batch: dict, batch_idx):
464464
"""Run validation step"""
465465
# Make all -1 values 0.0
466466
batch[self._target_key] = batch[self._target_key].clamp(min=0.0)
467-
print(f"batch shape: {batch[self._target_key].shape}")
468-
print(f"Shape after slice: {batch[self._target_key][:, -self.forecast_len_30 :, 0].shape}")
469467
y_hat = self(batch)
470468
# Sensor seems to be in batch, station, time order
471-
print(f"y_hat shape: {y_hat.shape}")
472-
print(f"batch shape: {batch[self._target_key].shape}")
473-
print(f"Shape after slice: {batch[self._target_key][:, -self.forecast_len_30 :, 0].shape}")
474-
exit()
475469
y = batch[self._target_key][:, -self.forecast_len_30 :, 0]
476470

477471
losses = self._calculate_common_losses(y, y_hat)

0 commit comments

Comments
 (0)