change workflow

ajithvcoder · Nov 3, 2024 · 1f3421e · 1f3421e
1 parent 9e74bc9
commit 1f3421e
Showing 1 changed file with 28 additions and 28 deletions.
diff --git a/.github/workflows/ec2-pipeline.yml b/.github/workflows/ec2-pipeline.yml
@@ -2,35 +2,35 @@ name: CML-EC2-Runner
 on: 
   workflow_dispatch:
 jobs:
-    # launch-runner:
-    #   runs-on: ubuntu-latest
-    #   steps:
-    #     - uses: actions/checkout@v3
-    #     - uses: iterative/setup-cml@v1
-    #     - name: Deploy runner on AWS EC2
-    #       env:
-    #         REPO_TOKEN: ${{ secrets.PERSONAL_ACCESS_TOKEN }}
-    #         AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID }}
-    #         AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
-    #       run: |
-    #         cml runner launch \
-    #         --cloud=aws \
-    #         --name=session-08 \
-    #         --cloud-region=ap-south-1 \
-    #         --cloud-type=g4dn.xlarge \
-    #         --cloud-hdd-size=64 \
-    #         --cloud-spot \
-    #         --single \
-    #         --labels=cml-gpu \
-    #         --idle-timeout=100
+    launch-runner:
+      runs-on: ubuntu-latest
+      steps:
+        - uses: actions/checkout@v3
+        - uses: iterative/setup-cml@v1
+        - name: Deploy runner on AWS EC2
+          env:
+            REPO_TOKEN: ${{ secrets.PERSONAL_ACCESS_TOKEN }}
+            AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID }}
+            AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
+          run: |
+            cml runner launch \
+            --cloud=aws \
+            --name=session-08 \
+            --cloud-region=ap-south-1 \
+            --cloud-type=g4dn.xlarge \
+            --cloud-hdd-size=64 \
+            --cloud-spot \
+            --single \
+            --labels=cml-gpu \
+            --idle-timeout=100
   train-and-report:
-    # runs-on: [self-hosted, cml-gpu]
-    # needs: launch-runner
-    # timeout-minutes: 20
-    # container:
-    #   image: docker://iterativeai/cml:0-dvc2-base1-gpu
-    #   options: --gpus all
-    runs-on: ubuntu-latest
+    runs-on: [self-hosted, cml-gpu]
+    needs: launch-runner
+    timeout-minutes: 20
+    container:
+      image: docker://iterativeai/cml:0-dvc2-base1-gpu
+      options: --gpus all
+    # runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
         with: