patch for modulelist bug

lucidrains · Feb 24, 2021 · 6c56dfb · 6c56dfb
1 parent e10dd18
commit 6c56dfb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/geometric_vector_perceptron/geometric_vector_perceptron.py b/geometric_vector_perceptron/geometric_vector_perceptron.py
@@ -118,8 +118,8 @@ def __init__(self, feats_x_in, vectors_x_in,
         self.vectors_edge_out = vectors_edge_out # N vectors features in output
         # aux layers
         self.vector_dim = vector_dim
-        self.norm = [GVPLayerNorm(self.feats_x_out), # + self.feats_edge_out
-                     GVPLayerNorm(self.feats_x_out)]
+        self.norm = nn.ModuleList([GVPLayerNorm(self.feats_x_out), # + self.feats_edge_out
+                                   GVPLayerNorm(self.feats_x_out)])
         self.dropout = GVPDropout(dropout)
         # this receives the vec_in message AND the receiver node
         self.W_EV = nn.Sequential(GVP(

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'geometric-vector-perceptron',
   packages = find_packages(),
-  version = '0.0.7',
+  version = '0.0.8',
   license='MIT',
   description = 'Geometric Vector Perceptron - Pytorch',
   author = 'Phil Wang and Eric Alcaide',