Added config

htahir1 · htahir1 · commit ded7bd66b234 · 2024-08-07T14:30:01.000+02:00
diff --git a/airflow-cloud-composer-etl-feature-train/configs/training_production.yaml b/airflow-cloud-composer-etl-feature-train/configs/training_production.yaml
@@ -23,4 +23,6 @@ steps:
     step_operator: gcp_airflow_vertex
     settings:
       step_operator.vertex:
-        machine_type: "n1-highmem-8"
+        accelerator_type: "NVIDIA_TESLA_P100" 
+        accelerator_count: 1
+        machine_type: "n1-standard-8"
