Hugging Face's logo

tomz616
/
pk

Model card Files
xet
 Community
pk
File size: 1,654 Bytes
bb1f598
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
---
default:
  generator: rdn
  feature_extractor: false
  discriminator: false
  training_set: div2k
  test_set: sample
log_dirs:
  logs: ./logs
  weights: ./weights
feature_extractor:
  vgg19:
    layers_to_extract:
      - 5
      - 9
generators:
  rrdn:
    C: 4
    D: 3
    G: 32
    G0: 32
    T: 4
    x: 4
  rdn:
    C: 6
    D: 20
    G: 64
    G0: 64
    x: 2
loss_weights:
  generator: 1.0
  feature_extractor: 0.0833
  discriminator: 0.01
losses:
  generator: mae
  discriminator: binary_crossentropy
  feature_extractor: mse
session:
  prediction:
    patch_size:
  training:
    steps_per_epoch: 1000
    patch_size: 32
    batch_size: 16
    epochs: 300
    n_validation_samples: 100
    learning_rate:
      initial_value: 0.0004
      decay_frequency: 50
      decay_factor: 0.5
    fallback_save_every_n_epochs: 2
    flatness:
      min: 0.0
      increase_frequency: null
      increase: 0.0
      max: 0.0
    metrics:
      generator: PSNR_Y
    monitored_metrics:
      val_loss: min
      val_PSNR_Y: max
    adam_optimizer:
      beta1: 0.9
      beta2: 0.999
      epsilon: null
test_sets:
  sample: ./data/input/sample
training_sets:
  custom data:
    lr_train_dir: ./data/custom/lr/train
    hr_train_dir: ./data/custom/hr/train
    lr_valid_dir: ./data/custom/lr/validation
    hr_valid_dir: ./data/custom/hr/validation
    data_name: custom
  div2k:
    lr_train_dir: ./data/DIV2K/DIV2K_train_LR_bicubic/X2
    hr_train_dir: ./data/DIV2K/DIV2K_train_HR
    lr_valid_dir: ./data/DIV2K/DIV2K_valid_LR_bicubic/X2
    hr_valid_dir: ./data/DIV2K/DIV2K_valid_HR
    data_name: div2k
weights_paths:
  discriminator:
  generator: