test3

guillaume-chervet · Sep 4, 2024 · 3d195ad · 3d195ad
1 parent 53385a7
commit 3d195ad
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -209,11 +209,11 @@ jobs:
           cd model
           az ml model list --name cats-dogs-others --resource-group ${{ env.AZURE_RESOURCE_GROUP_NAME }} --workspace-name ${{ env.AZURE_ML_WORKSPACE_NAME }}
           echo "here3 $EXPERIMENT_ID $MODEL_VERSION"
+          $DOWNLOAD_PATH = $PWD
           cd ..
           cd train
           chmod +x ./run_download_model.sh
-          ./run_download_model.sh $MODEL_VERSION ${{ env.AZURE_RESOURCE_GROUP_NAME }} ${{ env.AZURE_ML_WORKSPACE_NAME }} ${{ env.AZURE_LOCATION }} ${{ secrets.AZURE_SUBSCRIPTION_ID }}
-          
+          ./run_download_model.sh $MODEL_VERSION ${{ env.AZURE_RESOURCE_GROUP_NAME }} ${{ env.AZURE_ML_WORKSPACE_NAME }} ${{ env.AZURE_LOCATION }} ${{ secrets.AZURE_SUBSCRIPTION_ID }} $DOWNLOAD_PATH
           
           echo "az ml model download --name cats-dogs-others --version $MODEL_VERSION --resource-group ${{ env.AZURE_RESOURCE_GROUP_NAME }} --workspace-name ${{ env.AZURE_ML_WORKSPACE_NAME }}"
           #az ml model download --name cats-dogs-others --version $MODEL_VERSION --resource-group ${{ env.AZURE_RESOURCE_GROUP_NAME }} --workspace-name ${{ env.AZURE_ML_WORKSPACE_NAME }}

diff --git a/train/azureml_run_download_model.py b/train/azureml_run_download_model.py
@@ -11,13 +11,15 @@
 parser.add_argument("--workspace_name", type=str)
 parser.add_argument("--location", type=str)
 parser.add_argument("--version", type=str, default="1")
+parser.add_argument("--download_path", type=str, default="./")
 
 args = parser.parse_args()
 subscription_id = args.subscription_id
 resource_group_name = args.resource_group_name
 workspace_name = args.workspace_name
 location = args.location
 version = args.version
+download_path = args.download_path
 
 try:
     credential = DefaultAzureCredential()
@@ -38,6 +40,5 @@
 )
 
 model_name = 'cats-dogs-others'
-download_path = './'
 
 ml_client.models.download(name=model_name, version=version, download_path=download_path)
diff --git a/train/run_download_model.sh b/train/run_download_model.sh
@@ -3,13 +3,15 @@ AZURE_RESOURCE_GROUP_NAME=$2
 AZURE_ML_WORKSPACE_NAME=$3
 AZURE_LOCATION=$4
 AZURE_SUBSCRIPTION_ID=$5
+DOWNLOAD_PATH=$6
 
 poetry run python azureml_run_download_model.py \
     --subscription_id "$AZURE_SUBSCRIPTION_ID" \
     --resource_group_name "$AZURE_RESOURCE_GROUP_NAME" \
     --workspace_name "$AZURE_ML_WORKSPACE_NAME" \
     --location "$AZURE_LOCATION" \
-    --version "$MODEL_VERSION"
+    --version "$MODEL_VERSION" \
+    --download_path "$DOWNLOAD_PATH"
 
 #az ml model download --name cats-dogs-others  \
 #    --version "$MODEL_VERSION"  \