Modified docstrings of transformer drop-in replacements.

Nix-da · Jan 31, 2022 · f56f929 · f56f929
1 parent d6d88c8
commit f56f929
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/hflayers/transformer.py b/hflayers/transformer.py
@@ -99,6 +99,9 @@ def output_size(self) -> int:
 
 
 class HopfieldDecoderLayer(Module):
+    """
+    Module with underlying Hopfield associations to be used as a decoder in transformer-like architectures.
+    """
 
     def __init__(self,
                  hopfield_association_self: Hopfield,

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name=r'hopfield-layers',
-    version=r'1.0.1',
+    version=r'1.0.2',
     author=r'Bernhard Schäfl',
     author_email=r'schaefl@ml.jku.at',
     url=r'https://github.com/ml-jku/hopfield-layers',