In train.yaml remove seq length and epoch arguments (redo)

matsuobasho · Oct 12, 2023 · 1a2dad3 · 1a2dad3
1 parent 6aef523
commit 1a2dad3
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/codegen_model_comparison/cloud/train.yaml b/codegen_model_comparison/cloud/train.yaml
@@ -10,27 +10,21 @@ inputs:
     type: uri_file
   batch_size:
     type: integer
-  seq_length:
-    type: integer
-  epochs:
-    type: integer
   learning_rate:
     type: number
 
 outputs:
   model_output:
     type: mlflow_model
 
-code: ./src
+code: ../src
 
-environment: azureml:env_finetune_component:@latest
+environment: azureml:AzureML-sklearn-1.0-ubuntu20.04-py38-cpu@latest
 
 command: >-
   python train.py
-    --data_path ${{inputs.data}}
+    --data_path ${{inputs.data_path}}
     --output_path ${{outputs.model_output}}
     --batch_size ${{inputs.batch_size}}
-    --seq_length ${{inputs.seq_length}}
-    --epochs ${{inputs.epochs}}
     --learning_rate ${{inputs.learning_rate}}