-
Notifications
You must be signed in to change notification settings - Fork 0
/
conf_resnet.yaml
95 lines (94 loc) · 1.86 KB
/
conf_resnet.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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
speckle:
nscreens: 8
hArray: [0.1, 300, 650, 1e3, 2e3, 5e3, 10e3, 20e3, 40e3]
splits: [150, 475, 825, 1500, 3500, 7500, 15000, 30000]
lambda: 550
original_res: 105
datadirectory: tests/test_data/speckles
preproc:
polresize: 105
polarize: false
equivariant: false
randomrotate: false
centercrop: -1
resize: 105
speckreps: 1
ensemble: 1
ensemble_unif: false
normalization: lin
img_normalization: false
dataname: all_images_resnet.pt
J_details: true
Fried_details: true
Scintillation_w_details: true
Isoplanatic_details: true
noise:
D: 1 # Fraction of field
t: 0.01 # Width spider
snr: 15 # Signal-to-noise ratio
dT: 0.2 # Telescope diameter
dO: 0.44 # fraction obscuration
rn: 3.5 # random noise
fw: 32e3 # full well capacity
bit: 12 # bit level (sample depth)
discretize: off # Whether to discretize the output ('on' or 'off').
model:
name: Model_test_resnet
type: resnet18
signal_noise_r: 37/6
save_every: 10
pretrained: true
hyppar:
maxepochs: 3
ttsplit: 0.2
batch_size: 16
lr: 0.0001
lr_scheduler: Plateau
early_stopping: 50
optimizer: Adam
loss:
MAE: 0
MSE: 1
JMAE: 0
JMSE: 0
Cn2MAE: 0
Cn2MSE: 0
Pearson: 0
Fried: 0
Isoplanatic: 0
Rytov: 0
Scintillation_w: 0
Scintillation_ms: 0
val_loss:
Fried: 1
scnn:
KERNEL_SIZES: [9, 7, 7]
PADDINGS: [2, 1, 1]
STRIDES: [2, 1, 1]
FEATURE_FIELDS: [16, 32, 32]
POOL_INDICES: [2]
SIGMA: [0.66]
POOL_STRIDES: [2]
POOL_PADDINGS: [0]
final_block:
- type: Dropout
p: 0.2
- type: Linear
out_features: 1024
- type: BatchNorm1d
num_features: 1024
- type: ELU
- type: Dropout
p: 0.1
- type: Linear
in_features: 1024
out_features: 1024
- type: BatchNorm1d
num_features: 1024
- type: ELU
- type: Dropout
p: 0.1
- type: Linear
in_features: 1024
out_features: nscreens
- type: Sigmoid