From 9a96154b0c0d58889e1babdc8be605afa584c8cb Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Sat, 13 Mar 2021 16:38:22 -0600
Subject: [PATCH] Adding ImagNet tests to Keras automated testing script
---
hpvm/projects/keras/scripts/test_benchmarks.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/hpvm/projects/keras/scripts/test_benchmarks.py b/hpvm/projects/keras/scripts/test_benchmarks.py
index 201dc5db47..3ac145b9dd 100644
--- a/hpvm/projects/keras/scripts/test_benchmarks.py
+++ b/hpvm/projects/keras/scripts/test_benchmarks.py
@@ -160,20 +160,26 @@ if __name__ == "__main__":
testMgr = BenchmarkTests()
AlexNet = Benchmark("src/alexnet.py", 79.28)
+ AlexNet_ImageNet = Benchmark("src/alexnet_imagenet.py", 56.30)
AlexNet2 = Benchmark("src/alexnet2.py", 84.98)
LeNet = Benchmark("src/lenet.py", 98.70)
MobileNet = Benchmark("src/mobilenet_cifar10.py", 84.42)
ResNet18 = Benchmark("src/resnet18_cifar10.py", 89.56)
+ ResNet50 = Benchmark("src/resnet50_imagenet.py", 75.10)
VGG16_cifar10 = Benchmark("src/vgg16_cifar10.py", 89.96)
VGG16_cifar100 = Benchmark("src/vgg16_cifar100.py", 66.50)
+ VGG16_ImageNet = Benchmark("src/vgg16_imagenet.py", 69.46)
testMgr.addBenchmark(AlexNet)
+ testMgr.addBenchmark(AlexNet_ImageNet)
testMgr.addBenchmark(AlexNet2)
testMgr.addBenchmark(LeNet)
testMgr.addBenchmark(MobileNet)
testMgr.addBenchmark(ResNet18)
+ testMgr.addBenchmark(ResNet50)
testMgr.addBenchmark(VGG16_cifar10)
testMgr.addBenchmark(VGG16_cifar100)
+ testMgr.addBenchmark(VGG16_ImageNet)
testMgr.runKerasTests()
testMgr.printKerasSummary()
--
GitLab