From 328fa40c0c7c9dfff3b6f7cb8b72f5308c83dbf1 Mon Sep 17 00:00:00 2001
From: nz11 <nz11@illinois.edu>
Date: Sun, 22 Nov 2020 17:39:58 -0600
Subject: [PATCH] Update mobilenet_imagenet.py

---
 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 c5cef5193b..f2faaa8c2c 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)
+translate_to_approxhpvm(model, OUTPUT_DIR, X_test[:VAL_SIZE], y_true[:VAL_SIZE], 1000, dump_weights=False)
 
 dumpCalibrationData(OUTPUT_DIR + 'test_input.bin', X_test, OUTPUT_DIR + 'test_labels.bin', y_true)
 
-- 
GitLab