diff --git a/llvm/projects/keras/src/mobilenet_imagenet.py b/llvm/projects/keras/src/mobilenet_imagenet.py index c5cef5193b203633015a5dc8f4be065991bf5608..f2faaa8c2cff7ea0753855580b4e0fce57e64a4d 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)