data-science/config.yml at master · supertopdev/data-science · 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
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
---
default:
  generator: rdn
  feat_ext: false
  discriminator: false
  training_set: custom
  test_set: sample
dirs:
  logs: "./logs"
  weights: "./weights"
feat_ext:
  vgg19:
    layers_to_extract:
    - 5
    - 9
generators:
  rrdn:
    C: 4
    D: 3
    G: 64
    G0: 64
    T: 4
    x: 4
  rdn:
    C: 6
    D: 20
    G: 64
    G0: 64
    x: 2
loss_weights:
  MSE: 1
  feat_extr: 0.0833
  discriminator: 0.01
session:
  prediction:
    patch_size:
  training:
    steps_per_epoch: 500
    patch_size: 40
    batch_size: 16
    epochs: 300
    n_validation_samples: 100
    lr_decay_frequency: 30
    lr_decay_factor: 0.5
    learning_rate: 0.0004
    fallback_save_every_n_epochs: 2
    T: 0.1
    monitored_metrics:
      val_loss: 'min'
test_sets:
  sample: "./data/input/sample"
training_sets:
  custom data:
    training_input: "./data/custom/lr/train"
    training_labels: "./data/custom/hr/train"
    validation_input: "./data/custom/lr/validation"
    validation_labels: "./data/custom/hr/validation"
    data_name: custom
weights_paths:
  discriminator:
  generator: weights/sample_weights/rdn-C6-D20-G64-G064-x2_enhanced-e219.hdf5