diff --git a/llvm/projects/keras/src/resnet50_imagenet.py b/llvm/projects/keras/src/resnet50_imagenet.py
index 689709d27595ae78ad46f7b356c179f10f534ef2..55d0918b7a526ba1a9866d2d8e3b0e2e8608cc25 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)