...

pbloem · Mar 31, 2019 · c0cc61c · c0cc61c
1 parent b6319e5
commit c0cc61c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/layers.py b/layers.py
@@ -198,7 +198,7 @@ def forward(self, vxin, hxin, h):
             hx = hx + self.tohori(vx)
 
         if self.gates:
-            vx = self.gate(vx, h,  (self.vvf, self.vvg))
+            vx = self.gate(vx, h, (self.vvf, self.vvg))
             hx = self.gate(hx, h, (self.vhf, self.vhg))
 
         if self.res_connection:

diff --git a/models.py b/models.py
@@ -3,6 +3,10 @@
 from layers import *
 
 class Gated(nn.Module):
+    """
+    Model combining several gated pixelCNN layers with a conditional input (usually the class)
+
+    """
 
     def __init__(self, input_size, conditional_size, channels, num_layers, k=7, padding=3):
         super().__init__()