diff --git a/llvm/projects/keras/src/vgg16_imagenet.py b/llvm/projects/keras/src/vgg16_imagenet.py index 91e856d5652585caaef4cfecf78024fdbcd8befb..b41df8b83a966495d6a1c8281745e91181a66c41 100644 --- a/llvm/projects/keras/src/vgg16_imagenet.py +++ b/llvm/projects/keras/src/vgg16_imagenet.py @@ -217,7 +217,7 @@ print ('test size', len(X_test)) -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)