forked from Azure/azureml-examples
-
Notifications
You must be signed in to change notification settings - Fork 0
/
job.yml
19 lines (19 loc) · 618 Bytes
/
job.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
$schema: https://azuremlschemas.azureedge.net/latest/commandJob.schema.json
code: src
command: >-
pip install -r requirements.txt &&
python main.py
--iris-csv ${{inputs.iris_csv}}
--epochs ${{inputs.epochs}}
--lr ${{inputs.lr}}
inputs:
iris_csv:
type: uri_file
path: https://azuremlexamples.blob.core.windows.net/datasets/iris.csv
epochs: 10
lr: 0.1
environment: azureml:AzureML-sklearn-0.24-ubuntu18.04-py37-cpu@latest
compute: azureml:cpu-cluster
display_name: pytorch-iris-example
experiment_name: pytorch-iris-example
description: Train a neural network with PyTorch on the Iris dataset.