-
Notifications
You must be signed in to change notification settings - Fork 1
/
MLproject
34 lines (30 loc) · 1.1 KB
/
MLproject
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
# Author: Luis Kuhn Cuellar
# Email: luis.kuhn@gmail.com
# Version: 0.1.0-SNAPSHOT
# License: MIT
name: biomedical_image_segmentation
# conda_env: environment.yml
docker_env:
image: ghcr.io/luiskuhn/biomedical_image_segmentation:0.1.0-SNAPSHOT
volumes: ["${PWD}/data:/data"]
entry_points:
main:
parameters:
cuda: {type: string, default: 'True'}
epochs: {type: int, default: 5}
general-seed: {type: int, default: 0}
pytorch-seed: {type: int, default: 0}
training-batch-size: {type: int, default: 64}
test-batch-size: {type: int, default: 1000}
learning-rate: {type: float, default: 0.01}
log-interval: {type: int, default: 100}
command: |
python biomedical_image_segmentation/biomedical_image_segmentation.py \
--cuda {cuda} \
--epochs {epochs} \
--general-seed {general-seed} \
--pytorch-seed {pytorch-seed} \
--training-batch-size {training-batch-size} \
--test-batch-size {test-batch-size} \
--learning-rate {learning-rate} \
--log-interval {log-interval}