fix(cuboid_transformer): Modified the argument order of '_generalize_…

…padding' so that 'pad_h' comes before 'pad_w' (#48)
amazon-science · May 25, 2023 · a9b42c1 · a9b42c1
1 parent e60ff41
commit a9b42c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/earthformer/cuboid_transformer/cuboid_transformer.py b/src/earthformer/cuboid_transformer/cuboid_transformer.py
@@ -277,7 +277,7 @@ def forward(self, x):
             T += pad_t
             H += pad_h
             W += pad_w
-            x = _generalize_padding(x, pad_t, pad_w, pad_h, padding_type=self.padding_type)
+            x = _generalize_padding(x, pad_t, pad_h, pad_w, padding_type=self.padding_type)
 
         x = x.reshape((B,
                        T // self.downsample[0], self.downsample[0],