-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdvc.yaml
65 lines (65 loc) · 2.13 KB
/
dvc.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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
stages:
extract_data:
cmd: python3 src/data/extract_data.py
deps:
- src/data/extract_data.py
- data/raw/datasets.zip
outs:
- data/raw/datasets:
cache: false
make_dataset:
cmd: python3 src/data/make_dataset.py
deps:
- data/raw/datasets
- src/data/make_dataset.py
outs:
- data/processed/datasets_processed.zip:
persist: true
train_model:
cmd: python3 src/models/train_model.py
deps:
- data/processed/datasets_processed.zip
- src/models/train_model.py
outs:
- models/saved/10_25_0.001_0.25_unet_model.pth:
persist: true
- models/saved/10_25_0.0001_0.25_unet_model.pth:
persist: true
- models/saved/10_25_0.0005_0.25_unet_model.pth:
persist: true
- models/saved/10_50_0.001_0.25_unet_model.pth:
persist: true
- models/saved/10_50_0.0001_0.25_unet_model.pth:
persist: true
- models/saved/10_50_0.0005_0.25_unet_model.pth:
persist: true
predict_model:
cmd: python3 src/models/predict_model.py
deps:
- models/saved/10_25_0.001_0.25_unet_model.pth
- models/saved/10_25_0.0001_0.25_unet_model.pth
- models/saved/10_25_0.0005_0.25_unet_model.pth
- models/saved/10_50_0.001_0.25_unet_model.pth
- models/saved/10_50_0.0001_0.25_unet_model.pth
- models/saved/10_50_0.0005_0.25_unet_model.pth
- src/models/predict_model.py
outs:
- models/saved/sd_10_25_0.001_0.25_unet_model.pth:
persist: true
- models/saved/sd_10_25_0.0001_0.25_unet_model.pth:
persist: true
- models/saved/sd_10_25_0.0005_0.25_unet_model.pth:
persist: true
- models/saved/sd_10_50_0.001_0.25_unet_model.pth:
persist: true
- models/saved/sd_10_50_0.0001_0.25_unet_model.pth:
persist: true
- models/saved/sd_10_50_0.0005_0.25_unet_model.pth:
persist: true
metrics:
- models/metrics/10_25_0.001_0.25.metrics
- models/metrics/10_25_0.0001_0.25.metrics
- models/metrics/10_25_0.0005_0.25.metrics
- models/metrics/10_50_0.001_0.25.metrics
- models/metrics/10_50_0.0001_0.25.metrics
- models/metrics/10_50_0.0005_0.25.metrics