fix: 🐛 Fix AttentiveScalenet for 2D

janelia-cellmap · Apr 29, 2024 · a4a1ef9 · a4a1ef9
1 parent 701cb24
commit a4a1ef9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/leibnetz/nets/attentive_scalenet.py b/src/leibnetz/nets/attentive_scalenet.py
@@ -103,6 +103,7 @@ def build_subnet(
                 gating_nc=base_nc * nc_increase_factor ** (i + 1),
                 input_nc=base_nc * nc_increase_factor**i,
                 identifier=output_key,
+                ndims=len(kernel_sizes[0]),
             )
         )
         input_key = output_key