Modify compute in pipeline.yaml

matsuobasho · Oct 12, 2023 · 811cef3 · 811cef3
1 parent 1683052
commit 811cef3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/codegen_model_comparison/cloud/pipeline.yaml b/codegen_model_comparison/cloud/pipeline.yaml
@@ -3,7 +3,7 @@ type: pipeline
 display_name: codegen_sweep
 description: Tune hyperparameters
 settings:
-  default_compute: azureml:roma2
+  default_compute: azureml:gpu-buck50
 jobs:
   sweep_step:
     type: sweep
@@ -12,7 +12,7 @@ jobs:
         type: uri_file
         path: azureml:code_train_data:1
       seq_length: 100
-      epochs: 3
+      epochs: 1
     outputs:
       model_output:
     sampling_algorithm: random