Print batch

openclimatefix · Dec 11, 2023 · a63dc94 · a63dc94
1 parent 272ce35
commit a63dc94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pvnet/utils.py b/pvnet/utils.py
@@ -250,7 +250,7 @@ def _get_numpy(key):
     y_id_key = BatchKey.gsp_id if key_to_plot == "gsp" else BatchKey.sensor_id
     t0_idx_key = BatchKey.gsp_t0_idx if key_to_plot == "gsp" else BatchKey.sensor_t0_idx
     time_utc_key = BatchKey.gsp_time_utc if key_to_plot == "gsp" else BatchKey.sensor_time_utc
-
+    print(batch)
     y = batch[y_key].cpu().numpy()
     y_hat = y_hat.cpu().numpy()
     gsp_ids = batch[y_id_key].cpu().numpy().squeeze()