fix resnet50 typo

uoguelph-mlrg · Jul 12, 2017 · b4de57b · b4de57b
1 parent 008b99e
commit b4de57b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/theanompi/models/lasagne_model_zoo/resnet50.py b/theanompi/models/lasagne_model_zoo/resnet50.py
@@ -345,7 +345,7 @@ def build_model(self):
         self.y = T.lvector('y')
         self.lr = T.scalar('lr')
 
-        net = build_model_resnet152(input_shape=(None, 3, 224, 224))
+        net = build_model_resnet50(input_shape=(None, 3, 224, 224))
 
         self.output_layer = net['prob']