MLproject

# Author: Carolin Schwitalla
# Email: carolin-schwitalla@web.de
# Version: 0.1.0-SNAPSHOT
# License: MIT


name: cell-nuclei-segmentation

# conda_env: environment.yml
docker_env:
  image: ghcr.io/caroamn/cell_nuclei_segmentation:0.1.0-SNAPSHOT
  volumes: ["${PWD}/data:/data"]
  environment: [["MLF_CORE_DOCKER_RUN", "TRUE"],["CUBLAS_WORKSPACE_CONFIG", ":4096:8"]]

entry_points:
  main:
    parameters:
      max_epochs: {type: int, default: 5}
      gpus: {type: int, default: 0}
      accelerator: {type str, default: "None"}
      training-batch-size: {type: int, default: 64}
      test-batch-size: {type: int, default: 1000}
      lr: {type: float, default: 0.01}
      general-seed: {type: int, default: 0}
      pytorch-seed: {type: int, default: 0}
      log-interval: {type: int, default: 100}
    command: |
          python cell_nuclei_segmentation/cell_nuclei_segmentation.py \
            --max_epochs {max_epochs} \
            --gpus {gpus} \
            --accelerator {accelerator} \
            --training-batch-size {training-batch-size} \
            --test-batch-size {test-batch-size} \
            --lr {lr} \
            --general-seed {general-seed} \
            --pytorch-seed {pytorch-seed} \
            --log-interval {log-interval}