-
Notifications
You must be signed in to change notification settings - Fork 1
/
dvc.lock
85 lines (85 loc) · 2.36 KB
/
dvc.lock
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
schema: '2.0'
stages:
filter_classes:
cmd: python src/data/filter_classes.py
deps:
- path: data/raw/
md5: 7193e3c79707fec0b257d8b86d08d415.dir
size: 288336418
nfiles: 11740
- path: src/data/filter_classes.py
md5: 7f80ac90b3a38336d22ef988cc5b4b8f
size: 1805
params:
params.yaml:
prepare.excluded_classes:
- None
outs:
- path: data/interim/filepath.csv
md5: 852d240d08ef0d5a1cebf690656e1d9c
size: 694261
- path: data/processed/labels_map.yaml
md5: 36eaa622a7d9f27d08b613608bf69fa5
size: 225
split_dataset:
cmd: python src/data/split_dataset.py
deps:
- path: data/interim/filepath.csv
md5: 852d240d08ef0d5a1cebf690656e1d9c
size: 694261
- path: src/data/split_dataset.py
md5: fd5c5165b62eda903619cead36643786
size: 1496
params:
params.yaml:
prepare.test_split: 0.2
prepare.val_split: 0.2
outs:
- path: data/processed/test.csv
md5: db543c86af5d5a39819f5ffeef257b2a
size: 138902
- path: data/processed/train.csv
md5: d60246a8df6be472370a13ea0982b318
size: 444164
- path: data/processed/val.csv
md5: b3cc7561eba69a1c023728703c5063e2
size: 111221
get_mean_std:
cmd: python src/data/get_img_mean_std.py
deps:
- path: data/interim/filepath.csv
md5: 852d240d08ef0d5a1cebf690656e1d9c
size: 694261
- path: src/data/get_img_mean_std.py
md5: 222dce76d1be4f63a9526008586b70bb
size: 1497
params:
params.yaml:
prepare.image_size: 227
outs:
- path: data/processed/image_mean_std.yaml
md5: fe14624dd253c18acac5ee984f8418d4
size: 140
train_model:
cmd: python src/models/train_model.py
deps:
- path: data/processed/image_mean_std.yaml
md5: fe14624dd253c18acac5ee984f8418d4
size: 140
- path: data/processed/test.csv
md5: db543c86af5d5a39819f5ffeef257b2a
size: 138902
- path: data/processed/train.csv
md5: d60246a8df6be472370a13ea0982b318
size: 444164
- path: data/processed/val.csv
md5: b3cc7561eba69a1c023728703c5063e2
size: 111221
- path: src/models/train_model.py
md5: 5660eb37c75bc7a7966afc1e662e2935
size: 6063
params:
params.yaml:
train.batch_size: 24
train.data_aug_policy: 0
train.learning_rate: 0.001