diff --git a/llvm/projects/keras/src/Benchmark.py b/llvm/projects/keras/src/Benchmark.py index 0d51cbb7e0c1c6434f17c644e6d25d9609140afb..e653e120dd9973fcf44a2b56fcca00c271bc9f81 100644 --- a/llvm/projects/keras/src/Benchmark.py +++ b/llvm/projects/keras/src/Benchmark.py @@ -35,6 +35,7 @@ class Benchmark: def run(self, argv): if len(argv) < 2: + print ("Usage: python ${benchmark.py} [hpvm_reload|keras_reload|train] [frontend]") sys.exit(0) model = self.buildModel() @@ -56,7 +57,7 @@ class Benchmark: if len(argv) > 2 and argv[2] == "frontend": if argv[1] == "keras_reload": - print("ERROR: Must load HPVM model to invoke frontend") + print("ERROR: Must load HPVM model to invoke frontend - use 'hpvm_reload'") sys.exit(1) translate_to_approxhpvm(model, self.hpvm_dir, X_test, Y_test, self.num_classes)