Sketchs_to_ColorImages/dvc.yaml at main · Karan-Choudhary/Sketchs_to_ColorImages · GitHub

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
stages:
  get_data:
    cmd: python src\get_data.py --config=params.yaml
    deps:
    - src\read_params.py
    - src\get_data.py
    - src\utils\preprocessing.py
    - data\train
    - data\val

  train_model:
    cmd: python src\train_model.py --config=params.yaml
    deps:
    - src\get_data.py
    - src\train_evaluate.py
    - src\display\showOutput.py
    - src\model\train.py
    - src\model\build.py
    - src\model\sample.py
    - src\model\loss\losses.py
    outs:
    - saved_models\generator\gen.h5
    - saved_models\discriminator\disc.h5

  evaluate_model:
    cmd: python src\evaluate.py --config=params.yaml
    deps:
    - src\get_data.py
    - src\read_params.py
    - src\display\showOutput.py
    outs:
    - Results