Debugging

openclimatefix · Dec 5, 2023 · 99aad17 · 99aad17
1 parent c3dc7ab
commit 99aad17
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pvnet/models/multimodal/linear_networks/basic_blocks.py b/pvnet/models/multimodal/linear_networks/basic_blocks.py
@@ -25,6 +25,9 @@ def __init__(
     def cat_modes(self, x):
         """Concatenate modes of input data into 1D feature vector"""
         if isinstance(x, OrderedDict):
+            # Print out all components of the OrderedDict
+            for key, value in x.items():
+                print(f"{key}: {value.shape}")
             return torch.cat([value for key, value in x.items()], dim=1)
         elif isinstance(x, torch.Tensor):
             return x