From 5acdd67f912f1ecd847045753b0bdfd0f2c68b2f Mon Sep 17 00:00:00 2001
From: nz11 <nz11@illinois.edu>
Date: Sun, 22 Nov 2020 17:41:37 -0600
Subject: [PATCH] translate without any training

---
 llvm/projects/keras/src/mobilenet_imagenet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/llvm/projects/keras/src/mobilenet_imagenet.py b/llvm/projects/keras/src/mobilenet_imagenet.py
index f2faaa8c2c..c5cef5193b 100644
--- a/llvm/projects/keras/src/mobilenet_imagenet.py
+++ b/llvm/projects/keras/src/mobilenet_imagenet.py
@@ -252,7 +252,7 @@ y_true = np.array(y_true)
 # model.fit_generator(generate(), steps_per_epoch=1000, validation_data=(X_test, to_categorical(y_true, num_classes=1000)), epochs=7)
 
 
-translate_to_approxhpvm(model, OUTPUT_DIR, X_test[:VAL_SIZE], y_true[:VAL_SIZE], 1000, dump_weights=False)
+translate_to_approxhpvm(model, OUTPUT_DIR, X_test[:VAL_SIZE], y_true[:VAL_SIZE], 1000)
 
 dumpCalibrationData(OUTPUT_DIR + 'test_input.bin', X_test, OUTPUT_DIR + 'test_labels.bin', y_true)
 
-- 
GitLab