train_cpu.yml

name: Train root-tissue-segmentation using CPU

on: [ push, pull_request ]

jobs:
    run:
        runs-on: ubuntu-latest
        strategy:
            matrix:
                python: [ 3.7, 3.8 ]

        steps:
            -   name: Reclaim space
                run: |
                    sudo rm -rf /usr/share/dotnet
                    sudo rm -rf /opt/ghc
            -   name: Check out source-code repository
                uses: actions/checkout@v2

            -   name: Build Docker container
                uses: docker/build-push-action@v1
                with:
                    dockerfile: Dockerfile
                    repository: ghcr.io/qbic-pipelines/root_tissue_segmentation
                    tags: 1.0.0 # <<MLF-CORE_FORCE_BUMP>>
                    push: false

            -   name: Setup Python
                uses: actions/setup-python@v1
                with:
                    python-version: ${{ matrix.python }}

            -   name: Install mlflow
                run: pip install mlflow

            -   name: Train on the CPU
                run: mlflow run . -P max_epochs=2 -P gpus=0