Skip to content

Fix/qpb-review

Fix/qpb-review #32

Workflow file for this run

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 . --build-image -A runtime=nvidia -P max_epochs=2 -P gpus=0