File size: 1,009 Bytes
3c0c5aa
 
 
 
 
 
 
 
 
 
0b86a0a
a711240
0b86a0a
7200298
 
0b86a0a
a711240
0b86a0a
 
7200298
 
5ad6755
 
 
a711240
 
 
7200298
 
 
a711240
 
 
dc053a0
 
5ad6755
 
 
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
stages:
  process_data:
    cmd: python3 src/code/make_dataset.py src/data/raw/nyu_depth_v2_labeled.mat src/data/raw/splits.mat
      src/data/processed
    deps:
    - src/code/make_dataset.py
    - src/data/raw/nyu_depth_v2_labeled.mat
    - src/data/raw/splits.mat
    outs:
    - src/data/processed/
  train:
    cmd: python3 src/code/training.py src/data/processed/train
    deps:
    - src/code/custom_data_loading.py
    - src/code/params.yml
    - src/code/training.py
    - src/data/processed/train
    outs:
    - src/models/
    - logs/train_params.yml:
        cache: false
    metrics:
    - logs/train_metrics.csv:
        cache: false
  eval:
    cmd: python3 src/code/eval.py src/data/processed/test
    deps:
    - src/code/params.yml
    - src/code/custom_data_loading.py
    - src/code/eval_metric_calculation.py
    - src/code/eval.py
    - src/models/model.pth
    - src/data/processed/test
    outs:
    - src/eval/examples/
    metrics:
    - logs/test_metrics.csv:
        cache: false