diff --git a/hpvm/projects/keras/scripts/test_benchmarks.py b/hpvm/projects/keras/scripts/test_benchmarks.py index 201dc5db4787ebcfe2fb80e6b1dc75cf0897eaad..3ac145b9dd0c0a7af0b6bbb9618eb41bc085cd6c 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()