-
Notifications
You must be signed in to change notification settings - Fork 1
/
config.yaml
66 lines (55 loc) · 1.23 KB
/
config.yaml
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
# General Settings
SR_type: 'MISR' # [SISR, MISR]
# dont forget to also set the sen2_amount
# Data Settings
Data:
# Loader Settings
train_batch_size: 12 #12
val_batch_size: 8
num_workers: 6
prefetch_factor: 2 #8 #2
# Detailed Dataset Setttings
dataset_type: 'SPOT6' # ['cv', 'SPOT6']
data_dir: 'E:/thesis_paper/data/' # ['data/openImages/',''E:/thesis_paper/data/']
sen2_amount: 4
spectral_matching: 'histogram'
return_type: 'cross_sensor' #interpolated_matched
# Model, Loss, Training Settings
Model:
load_checkpoint: True
ckpt_path: 'logs\curated_ckpts\pretrained_MISR_SPOT6synt.ckpt'
load_fusion_checkpoint: False
fusion_ckpt_path: 'logs\curated_ckpts\pretrained_Fusion.ckpt'
Warmup:
fusion: # train fusion in isolation previously
enable: True
epochs: 1
lr: 1e-4
Generator:
large_kernel_size: 9
small_kernel_size: 3
n_channels: 64
n_blocks: 16
scaling_factor: 4
Discriminator:
kernel_size: 3
n_channels: 64
n_blocks: 8
fc_size: 1024
TruncatedVGG:
i: 5
j: 4
Optimizers:
optim_g_lr: 1e-4
optim_d_lr: 1e-4
Schedulers:
metric: 'val/L1'
patience_g: 3
patience_d: 3
factor_g: 0.25
factor_d: 0.25
verbose: True
Losses:
adv_loss_beta: 1e-3
Logging:
num_val_images: 5