- fixed string

inferno-pytorch · Sep 1, 2017 · fb0d178 · fb0d178
1 parent 2a87874
commit fb0d178
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inferno/extensions/layers/reshape.py b/inferno/extensions/layers/reshape.py
@@ -9,7 +9,7 @@
            'As3D', 'As2D',
            'Concatenate', 'Cat',
            'ResizeAndConcatenate', 'PoolCat',
-           'Sum', 'Split']
+           'Sum', 'SplitChannels']
 
 
 class View(nn.Module):