-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.yaml
39 lines (38 loc) · 865 Bytes
/
config.yaml
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
35
36
37
38
lakefs:
s3_endpoint: ""
access_key: ""
secret_key: ""
data_repository: "smsc-basel"
branch: 'main/'
datasplit:
split: [70, 10, 20]
num_splits: 5
foldspath: "/tmp/folds"
mdgru:
featurefiles: ["flair.nii.gz", "t1.nii.gz", "t1ce.nii.gz", "t2.nii.gz"]
maskfiles: ["seg.nii.gz"]
datapath: "/tmp/mdgru"
locationtraining: "training.json"
locationvalidation: "validation.json"
locationtesting: "test.json"
optionname: "test_mdgru"
modelname: "test_mdgru_docker"
w: [240, 240]
p: [0, 0]
windowsizetesting: [240, 240]
paddingtesting: [0, 0]
iterations: 50000
validate_each: 50001
save_each: 5000
nclasses: 5
num_threads: 8
gpu: 0
cache_path: "/tmp/s3_cache"
use_s3: true
run:
container_name: "mdgru_s3"
docker_rootless: true
gpu_pool: [0]
detach_docker: true
cache_path: "/tmp/s3_cache"
cache_delete: true