updated

haowen-xu · May 18, 2020 · 50a06c9 · 50a06c9
1 parent a5854cf
commit 50a06c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorkit/layers/pixelcnn.py b/tensorkit/layers/pixelcnn.py
@@ -244,7 +244,7 @@ def __init__(self,
 
         if edge_bias:
             self.add_ones_channel = getattr(
-                conv_edge_bias, f'AddOnesChannel{spatial_ndims}d')()
+                edge_bias_conv_, f'AddOnesChannel{spatial_ndims}d')()
             in_channels += 1
         else:
             self.add_ones_channel = Identity()