Larger space kernels

openclimatefix · Jan 29, 2024 · 4dd7492 · 4dd7492
1 parent beab66e
commit 4dd7492
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pvnet/models/multimodal/encoders/encoders3d.py b/pvnet/models/multimodal/encoders/encoders3d.py
@@ -37,7 +37,7 @@ def __init__(
         super().__init__(sequence_length, image_size_pixels, in_channels, out_features)
 
         # Check that the output shape of the convolutional layers will be at least 1x1
-        cnn_spatial_output_size = image_size_pixels - 8 * number_of_conv3d_layers
+        cnn_spatial_output_size = image_size_pixels - 12 * number_of_conv3d_layers
         cnn_sequence_length = sequence_length - 4 * number_of_conv3d_layers
         if not (cnn_spatial_output_size >= 1):
             raise ValueError(
@@ -51,7 +51,7 @@ def __init__(
             nn.Conv3d(
                 in_channels=in_channels,
                 out_channels=conv3d_channels,
-                kernel_size=(5, 9, 9),
+                kernel_size=(7, 13, 13),
                 padding=(0, 0, 0),
             ),
             nn.ELU(),
@@ -61,7 +61,7 @@ def __init__(
                 nn.Conv3d(
                     in_channels=conv3d_channels,
                     out_channels=conv3d_channels,
-                    kernel_size=(5, 9, 9),
+                    kernel_size=(7, 13, 13),
                     padding=(0, 0, 0),
                 ),
                 nn.ELU(),