Skip to content

Commit

Permalink
update workflow settings
Browse files Browse the repository at this point in the history
  • Loading branch information
ajithvcoder committed Nov 3, 2024
1 parent e926ae2 commit 8b0500e
Showing 1 changed file with 11 additions and 2 deletions.
13 changes: 11 additions & 2 deletions .github/workflows/ec2-pipeline.yml
Original file line number Diff line number Diff line change
Expand Up @@ -21,11 +21,12 @@ jobs:
--cloud-hdd-size=64 \
--cloud-spot \
--single \
--labels=cml-gpu
--labels=cml-gpu \
--idle-timeout=100
train-and-report:
runs-on: [self-hosted, cml-gpu]
needs: launch-runner
timeout-minutes: 50400 # 35 days
timeout-minutes: 20
container:
image: docker://iterativeai/cml:0-dvc2-base1-gpu
options: --gpus all
Expand Down Expand Up @@ -61,3 +62,11 @@ jobs:
dvc pull -r myremote && \
dvc repro train
"
- name: Cleanup
if: always() # This ensures it runs even if previous steps fail
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 destroy --name=session-08 --cloud=aws

0 comments on commit 8b0500e

Please sign in to comment.