From 4f98a02c21c2f1864e1c3bde5b8c75fc767b3da2 Mon Sep 17 00:00:00 2001 From: ashishj1729 Date: Tue, 24 Dec 2024 13:25:50 +0530 Subject: [PATCH] Amending retrain-pipeline as per Model-deployment requirements 1. Model-path is deduced by modelversion 2. Only retrain-model will be present in Model.zip Change-Id: I0bc49b876b1d298b8d558e48f4c32c49fde2f0bf Signed-off-by: ashishj1729 --- kf-pipelines/qoe-pipeline-retrain-2.ipynb | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/kf-pipelines/qoe-pipeline-retrain-2.ipynb b/kf-pipelines/qoe-pipeline-retrain-2.ipynb index f933c35..f0ab77d 100644 --- a/kf-pipelines/qoe-pipeline-retrain-2.ipynb +++ b/kf-pipelines/qoe-pipeline-retrain-2.ipynb @@ -131,10 +131,7 @@ "\n", " # Load the model in SavedModel format\n", " # Path to the directory containing the saved model\n", - " if artifactversion ==\"1.0.0\":\n", - " model_path = \"./Model/1\"\n", - " else:\n", - " model_path= \"./Model/1/retrain\"\n", + " model_path = f\"./Model/{modelversion}\"\n", " \n", " model = tf.keras.models.load_model(model_path)\n", " model.compile(loss='mse', optimizer='adam', metrics=['mse'])\n", @@ -199,7 +196,7 @@ " print(updated_model_info)\n", " \n", " mm_sdk.upload_metrics(data, modelname, modelversion,new_artifactversion)\n", - " mm_sdk.upload_model(\"./\", modelname, modelversion, new_artifactversion)\n" + " mm_sdk.upload_model(\"./retrain/\", modelname, modelversion, new_artifactversion)\n" ] }, { -- 2.16.6