From 2cc00433e2f1c7baf8efa8f4873a905e096e05b5 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Sat, 13 Mar 2021 16:39:37 -0600
Subject: [PATCH] Fixing VGG16 ImageNet batch size
---
hpvm/projects/keras/src/vgg16_imagenet.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/hpvm/projects/keras/src/vgg16_imagenet.py b/hpvm/projects/keras/src/vgg16_imagenet.py
index 8c27714fcf..6b9458b537 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)
--
GitLab