Fixed the batch size for VGG16

richardboo · Jun 22, 2016 · ed4d964 · ed4d964
1 parent cd4ff8f
commit ed4d964
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/imagenet/models/helper.py b/examples/imagenet/models/helper.py
@@ -66,7 +66,7 @@ def std_spec(batch_size, isotropic=True):
     ResNet101: std_spec(batch_size=25),
     ResNet152: std_spec(batch_size=25),
     NiN: std_spec(batch_size=500),
-    VGG16: std_spec(batch_size=224)
+    VGG16: std_spec(batch_size=25)
 }