diff --git a/hpvm/projects/keras/src/vgg16_imagenet.py b/hpvm/projects/keras/src/vgg16_imagenet.py
index 8c27714fcf8799c5881e39a477c876cb2f0056a4..6b9458b5378c421f5ef8f8811e4721056fd19643 100644
--- a/hpvm/projects/keras/src/vgg16_imagenet.py
+++ b/hpvm/projects/keras/src/vgg16_imagenet.py
@@ -130,7 +130,7 @@ if __name__ == '__main__':
     data_dir = '' 
     src_dir = 'data/vgg16_imagenet_src/'
     num_classes = 1000
-    batch_size = 50
+    batch_size = 25
 
     alexnet = VGG16('VGG16_imagenet', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size)