diff --git a/.github/workflows/cml.yaml b/.github/workflows/cml.yaml new file mode 100644 index 0000000..1c23c8b --- /dev/null +++ b/.github/workflows/cml.yaml @@ -0,0 +1,23 @@ +name: mlworkflows +on: [push] +jobs: + run: + runs-on: [ubuntu-latest] + # optionally use a convenient Ubuntu LTS + CUDA + DVC + CML image + # container: docker://dvcorg/cml:0-dvc2-base1-gpu + # container: docker://ghcr.io/iterative/cml:0-dvc2-base1-gpu + steps: + - uses: actions/checkout@v2 + # may need to setup NodeJS & Python3 on e.g. self-hosted + # - uses: actions/setup-node@v2 + # with: + # node-version: '12' + # - uses: actions/setup-python@v2 + # with: + # python-version: '3.x' + - uses: iterative/setup-cml@v1 + - name: Train model + run: | + # Your ML workflow goes here + pip install -r requirements.txt + python train.py