Develop (#2)

* Increase dev version * Minor bugfix
jernsting · Aug 12, 2021 · 707246e · 707246e
1 parent bb66830
commit 707246e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/documentation/docs/layers/spatial_attention.md b/documentation/docs/layers/spatial_attention.md
@@ -8,7 +8,7 @@ Simple SpatialAttention for 2D filters.
 | Parameter | type | Description |
 | -----     | ----- | ----- |
 | in_channels | int | Number of input channels|
-| kernel_size | int, default=7 | Filter size vor CNN |
+| kernel_size | int, default=7 | Filter size for convolution filter |
 | batch_norm  | bool, default=True | If true batch normalization is applied |
 
 #### Return Value
@@ -36,7 +36,7 @@ Simple SpatialAttention for 3D filters.
 | Parameter | type | Description |
 | -----     | ----- | ----- |
 | in_channels | int | Number of input channels|
-| kernel_size | int, default=7 | Filter size for CNN |
+| kernel_size | int, default=7 | Filter size for convolution filter |
 | batch_norm  | bool, default=True | If true batch normalization is applied |
 
 #### Return Value

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
     use_setuptools()
     from setuptools import setup, find_packages
 
-__version__ = '0.0a0.dev1'
+__version__ = '0.0a0.dev2'
 
 with open("README.md", "r", encoding="utf-8") as fh:
     long_description = fh.read()

diff --git a/useful_layers/layers/spatial_attention.py b/useful_layers/layers/spatial_attention.py
@@ -54,6 +54,7 @@ def __init__(self,
         """
         self.in_channels = in_channels
         self.kernel_size = kernel_size
+        self.batch_norm = None
         if batch_norm:
             self.batch_norm = nn.BatchNorm2d
         self.conv = nn.Conv2d
@@ -79,6 +80,7 @@ def __init__(self,
         """
         self.in_channels = in_channels
         self.kernel_size = kernel_size
+        self.batch_norm = None
         if batch_norm:
             self.batch_norm = nn.BatchNorm3d
         self.conv = nn.Conv3d