Renamed variable Block in utils

jernsting · Jul 12, 2021 · e107e1c · e107e1c
1 parent 71338d9
commit e107e1c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/useful_layers/utils.py b/useful_layers/utils.py
@@ -14,16 +14,16 @@ def reduction_network(in_channels: int,
     if dim not in ["2d", "3d"]:
         raise ValueError(f'dim should be 2d or 3d. Got {dim}')
     if dim == "2d":
-        Block = nn.Conv2d
+        block = nn.Conv2d
     if dim == "3d":
-        Block = nn.Conv3d
-    conv1 = Block(in_channels=in_channels,
+        block = nn.Conv3d
+    conv1 = block(in_channels=in_channels,
                   out_channels=in_channels // reduction,
                   kernel_size=1,
                   stride=1,
                   padding=0,
                   bias=True)
-    conv2 = Block(in_channels=in_channels // reduction,
+    conv2 = block(in_channels=in_channels // reduction,
                   out_channels=in_channels,
                   kernel_size=1,
                   stride=1,