Skip to content

Commit c27538d

Browse files
authored
Updating YAML code for AML Extension tutorial
Updating the YAML config code examples for "Train an image classification TensorFlow model using the Azure Machine Learning Visual Studio Code Extension" tutorial. The schema for YAML config files have been updated so the code examples need to be updated for this documentation.
1 parent bf85676 commit c27538d

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

articles/machine-learning/tutorial-train-deploy-image-classification-model-vscode.md

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ The first thing you have to do to build an application in Azure Machine Learning
6161
$schema: https://azuremlschemas.azureedge.net/latest/workspace.schema.json
6262
name: TeamWorkspace
6363
location: WestUS2
64-
friendly_name: team-ml-workspace
64+
display_name: team-ml-workspace
6565
description: A workspace for training machine learning models
6666
tags:
6767
purpose: training
@@ -114,13 +114,11 @@ Like workspaces and compute targets, training jobs are defined using resource te
114114
115115
```yml
116116
$schema: https://azuremlschemas.azureedge.net/latest/commandJob.schema.json
117-
code:
118-
local_path: src
117+
code: src
119118
command: >
120119
python train.py
121-
environment: azureml:AzureML-TensorFlow2.4-Cuda11-OpenMpi4.1.0-py36:1
122-
compute:
123-
target: azureml:gpu-cluster
120+
environment: azureml:AzureML-tensorflow-2.4-ubuntu18.04-py37-cuda11-gpu:48
121+
compute: azureml:gpu-cluster
124122
experiment_name: tensorflow-mnist-example
125123
description: Train a basic neural network with TensorFlow on the MNIST dataset.
126124
```

0 commit comments

Comments
 (0)