diff --git a/net.py b/net.py index 630ad59..b971286 100644 --- a/net.py +++ b/net.py @@ -76,7 +76,7 @@ def addExtraLayers(): elif layerAct == 'maxout': newLayers.append({ 'type': 'maxout', - 'group_size': layer['group_size'] if group_size in layerKeys else 2 + 'group_size': layer['group_size'] if 'group_size' in layerKeys else 2 }) else: print 'Error: Unsupported activation' @@ -191,4 +191,4 @@ def fromJSON(self, json): elif layerType == 'mex': obj = MexLayer(layer) obj.fromJSON(layer) - self.layers.append(obj) \ No newline at end of file + self.layers.append(obj)