diff --git a/hpvm/test/dnn_benchmarks/keras/alexnet.py b/hpvm/test/dnn_benchmarks/keras/alexnet.py index d6e7f98c4375fcb603c1f9549ea931410d7423ad..849e1a267de4d9f9f647cc642f1b722eba46efa1 100644 --- a/hpvm/test/dnn_benchmarks/keras/alexnet.py +++ b/hpvm/test/dnn_benchmarks/keras/alexnet.py @@ -150,9 +150,9 @@ if __name__ == '__main__': batch_size = 500 # Batch Size set to 500 - Adjust this value based on your GPU memory # All Classes inherit from 'Benchmark` class in src/Benchmark.py and have a common Constructor - model = AlexNet_CIFAR10('AlexNet_CIFAR10', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) + AlexNet = AlexNet_CIFAR10('AlexNet_CIFAR10', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) # This invokes the common run function in src/Benchmark.py - model.exportToHPVM(sys.argv) + AlexNet.exportToHPVM(sys.argv) diff --git a/hpvm/test/dnn_benchmarks/keras/alexnet2.py b/hpvm/test/dnn_benchmarks/keras/alexnet2.py index 22b499d577a3b42866465614cca4d2fd71e87942..7a75d9b5b3159a23157e2babbede5726493039a3 100644 --- a/hpvm/test/dnn_benchmarks/keras/alexnet2.py +++ b/hpvm/test/dnn_benchmarks/keras/alexnet2.py @@ -142,6 +142,6 @@ if __name__ == '__main__': num_classes = 10 batch_size = 500 - model = AlexNet2_CIFAR10('AlexNet2_CIFAR10', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) + AlexNet2 = AlexNet2_CIFAR10('AlexNet2_CIFAR10', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) - model.exportToHPVM(sys.argv) + AlexNet2.exportToHPVM(sys.argv) diff --git a/hpvm/test/dnn_benchmarks/keras/alexnet_imagenet.py b/hpvm/test/dnn_benchmarks/keras/alexnet_imagenet.py index e23fef741715be9791c1aba0210a27ef5b947934..20e43edb073725d2ef7329f01fc0de26f0907792 100644 --- a/hpvm/test/dnn_benchmarks/keras/alexnet_imagenet.py +++ b/hpvm/test/dnn_benchmarks/keras/alexnet_imagenet.py @@ -99,9 +99,9 @@ if __name__ == '__main__': num_classes = 1000 batch_size = 50 - model = AlexNet('AlexNet_Imagenet', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) + AlexNet = AlexNet('AlexNet_Imagenet', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) - model.exportToHPVM(sys.argv) + AlexNet.exportToHPVM(sys.argv) diff --git a/hpvm/test/dnn_benchmarks/keras/lenet.py b/hpvm/test/dnn_benchmarks/keras/lenet.py index 83f76ef306c31c25d1f1e961ab1922e89c217b28..840dfddfe2eefe6573d470d050965a0bf0fa3273 100644 --- a/hpvm/test/dnn_benchmarks/keras/lenet.py +++ b/hpvm/test/dnn_benchmarks/keras/lenet.py @@ -110,6 +110,6 @@ if __name__ == '__main__': print (reload_dir) - model = LeNet_MNIST('LeNet_MNIST', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) + LeNet = LeNet_MNIST('LeNet_MNIST', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) - model.exportToHPVM(sys.argv) + LeNet.exportToHPVM(sys.argv) diff --git a/hpvm/test/dnn_benchmarks/keras/mobilenet_cifar10.py b/hpvm/test/dnn_benchmarks/keras/mobilenet_cifar10.py index a236e0305f6f09bfa5d3335f405e6d3fb337894d..d59c85c558fd9963598861b60637efd7155ed570 100644 --- a/hpvm/test/dnn_benchmarks/keras/mobilenet_cifar10.py +++ b/hpvm/test/dnn_benchmarks/keras/mobilenet_cifar10.py @@ -188,7 +188,7 @@ if __name__ == '__main__': num_classes = 10 batch_size = 500 - model = MobileNet_CIFAR10('MobileNet_CIFAR10', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) + MobileNet = MobileNet_CIFAR10('MobileNet_CIFAR10', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) - model.exportToHPVM(sys.argv) + MobileNet.exportToHPVM(sys.argv) diff --git a/hpvm/test/dnn_benchmarks/keras/resnet18_cifar10.py b/hpvm/test/dnn_benchmarks/keras/resnet18_cifar10.py index 9e8c23cdc6df6bb168257fa69cfba6015e9569f2..e5d564b02dad1f8f3c23c7df20cb555aba733155 100644 --- a/hpvm/test/dnn_benchmarks/keras/resnet18_cifar10.py +++ b/hpvm/test/dnn_benchmarks/keras/resnet18_cifar10.py @@ -560,7 +560,7 @@ if __name__ == '__main__': num_classes = 10 batch_size = 500 - model = ResNet18_CIFAR10('ResNet18_CIFAR10', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) + ResNet18 = ResNet18_CIFAR10('ResNet18_CIFAR10', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) - model.exportToHPVM(sys.argv) + ResNet18.exportToHPVM(sys.argv) diff --git a/hpvm/test/dnn_benchmarks/keras/resnet50_imagenet.py b/hpvm/test/dnn_benchmarks/keras/resnet50_imagenet.py index 9ac71337b94899002296b736469d480083f7b7ad..94f6b291cbc8310bfa3bb1952ed56143b2040ef1 100644 --- a/hpvm/test/dnn_benchmarks/keras/resnet50_imagenet.py +++ b/hpvm/test/dnn_benchmarks/keras/resnet50_imagenet.py @@ -147,9 +147,9 @@ if __name__ == '__main__': num_classes = 1000 batch_size = 50 - model = ResNet50('ResNet50_imagenet', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) + ResNet50 = ResNet50('ResNet50_imagenet', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) - model.exportToHPVM(sys.argv) + ResNet50.exportToHPVM(sys.argv) diff --git a/hpvm/test/dnn_benchmarks/keras/vgg16_cifar10.py b/hpvm/test/dnn_benchmarks/keras/vgg16_cifar10.py index 03563cba8a79a58ef4cc9e04160f2ae6aaca2974..a3edd2d6b9f43ec9eaf5c218f3b50ca1dcb2c9ed 100644 --- a/hpvm/test/dnn_benchmarks/keras/vgg16_cifar10.py +++ b/hpvm/test/dnn_benchmarks/keras/vgg16_cifar10.py @@ -190,8 +190,8 @@ if __name__ == '__main__': num_classes = 10 batch_size = 500 - model = VGG16_CIFAR10('VGG16_CIFAR10', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) + VGG16 = VGG16_CIFAR10('VGG16_CIFAR10', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) - model.exportToHPVM(sys.argv) + VGG16.exportToHPVM(sys.argv) diff --git a/hpvm/test/dnn_benchmarks/keras/vgg16_cifar100.py b/hpvm/test/dnn_benchmarks/keras/vgg16_cifar100.py index 716b441706f26008aae9e30c96cad23fcd0b398e..b5f34f7c7dded4ab21bc5b8bc78712a9a8672f35 100644 --- a/hpvm/test/dnn_benchmarks/keras/vgg16_cifar100.py +++ b/hpvm/test/dnn_benchmarks/keras/vgg16_cifar100.py @@ -205,7 +205,7 @@ if __name__ == '__main__': num_classes = 100 batch_size = 100 - model = VGG16_CIFAR100('VGG16_CIFAR100', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) + VGG16 = VGG16_CIFAR100('VGG16_CIFAR100', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) - model.exportToHPVM(sys.argv) + VGG16.exportToHPVM(sys.argv) diff --git a/hpvm/test/dnn_benchmarks/keras/vgg16_imagenet.py b/hpvm/test/dnn_benchmarks/keras/vgg16_imagenet.py index bcba54717484b53687497dc471a1c257793bdf47..93cbf6f4785d0e24e9324f4b439f54d70e2d97f5 100644 --- a/hpvm/test/dnn_benchmarks/keras/vgg16_imagenet.py +++ b/hpvm/test/dnn_benchmarks/keras/vgg16_imagenet.py @@ -132,9 +132,9 @@ if __name__ == '__main__': num_classes = 1000 batch_size = 25 - alexnet = VGG16('VGG16_imagenet', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) + VGG16 = VGG16('VGG16_imagenet', reload_dir, keras_model_file, data_dir, src_dir, num_classes, batch_size) - alexnet.exportToHPVM(sys.argv) + VGG16.exportToHPVM(sys.argv)