at print statment

openclimatefix · Sep 5, 2024 · e227b65 · e227b65
1 parent a17ad48
commit e227b65
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py
@@ -623,6 +623,8 @@ def _log_validation_results(self, batch, y_hat, accum_batch_num):
             y_i = y[i].detach().cpu().numpy()
             y_hat_i = y_hat[i].detach().cpu().numpy()
 
+            print(BatchKey._member_map_)
+
             time_utc_key = BatchKey[f"{self._target_key}_time_utc"]
             time_utc = batch[time_utc_key][i, -self.forecast_len :].detach().cpu().numpy()