diff --git a/llvm/projects/keras/src/alexnet_imagenet.py b/llvm/projects/keras/src/alexnet_imagenet.py
index 1423b8c2e9c7bc02b32e31a7b4028872f8301aa5..41cf2d837cf8ba764b614512f40a0f6c0522ab3d 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)