From f3362b384b4cb4c6257225d4b77e90235082a9fa Mon Sep 17 00:00:00 2001
From: nz11 <nz11@illinois.edu>
Date: Sun, 22 Nov 2020 17:41:02 -0600
Subject: [PATCH] Update alexnet_imagenet.py

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

diff --git a/llvm/projects/keras/src/alexnet_imagenet.py b/llvm/projects/keras/src/alexnet_imagenet.py
index 1423b8c2e9..41cf2d837c 100644
--- a/llvm/projects/keras/src/alexnet_imagenet.py
+++ b/llvm/projects/keras/src/alexnet_imagenet.py
@@ -241,7 +241,7 @@ else:
 #     K.set_value(model.optimizer.lr, 0.000001)
 #     model.fit_generator(generate(), steps_per_epoch=1000, validation_data=(X_test, to_categorical(y_true, num_classes=1000)), epochs=3)
 
-translate_to_approxhpvm(model, OUTPUT_DIR, X_tune, y_tune, 1000, dump_weights=False)
+translate_to_approxhpvm(model, OUTPUT_DIR, X_tune, y_tune, 1000)
 
 # # dumpCalibrationData2(OUTPUT_DIR + 'test_input_10K.bin', X_test, OUTPUT_DIR + 'test_labels_10K.bin', y_true)
 # dumpCalibrationData2(OUTPUT_DIR + 'tune_input.bin', X_tune, OUTPUT_DIR + 'tune_labels.bin', y_tune)
-- 
GitLab