From 71e970f4094e8540756a344fc9547530e8be4326 Mon Sep 17 00:00:00 2001
From: nz11 <nz11@illinois.edu>
Date: Sun, 22 Nov 2020 17:42:15 -0600
Subject: [PATCH] translate without any training

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

diff --git a/llvm/projects/keras/src/resnet50_imagenet.py b/llvm/projects/keras/src/resnet50_imagenet.py
index 689709d275..55d0918b7a 100644
--- a/llvm/projects/keras/src/resnet50_imagenet.py
+++ b/llvm/projects/keras/src/resnet50_imagenet.py
@@ -274,7 +274,7 @@ else:
 #     model.fit_generator(generate(), steps_per_epoch=1000, validation_data=(X_test, to_categorical(y_true, num_classes=1000)), epochs=6)
 #     model.save_weights(OUTPUT_DIR + 'weights.h5')
 
-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