valentinsingularity commited on
Commit
57384df
·
verified ·
1 Parent(s): 7a5f454

Delete configs/a_base_point_neutron_conv_standard.yaml

Browse files
configs/a_base_point_neutron_conv_standard.yaml DELETED
@@ -1,144 +0,0 @@
1
- general:
2
- name: a_base_point_neutron_conv_standard
3
- root_dir: null
4
-
5
- dset:
6
- cls: ReflectivityDataLoader
7
- prior_sampler:
8
- cls: SubpriorParametricSampler
9
- kwargs:
10
- param_ranges:
11
- thicknesses: [1., 500.]
12
- roughnesses: [0., 60.]
13
- slds: [-8., 16.]
14
- bound_width_ranges:
15
- thicknesses: [1.0e-2, 500.]
16
- roughnesses: [1.0e-2, 60.]
17
- slds: [1.0e-2, 5.]
18
- model_name: standard_model
19
- max_num_layers: 2
20
- constrained_roughness: true
21
- max_thickness_share: 0.5
22
- logdist: false
23
- scale_params_by_ranges: false
24
- scaled_range: [-1., 1.]
25
- device: 'cuda'
26
-
27
- q_generator:
28
- cls: ConstantQ
29
- kwargs:
30
- q: [0.02, 0.15, 128]
31
- device: 'cuda'
32
-
33
- # q_generator:
34
- # cls: VariableQ
35
- # kwargs:
36
- # q_min_range: [0.005, 0.03]
37
- # q_max_range: [0.15, 0.3]
38
- # n_q_range: [128, 128]
39
- # device: 'cuda'
40
-
41
- intensity_noise:
42
- cls: GaussianExpIntensityNoise
43
- kwargs:
44
- relative_errors: [0.01, 0.3]
45
- consistent_rel_err: false
46
- apply_shift: false #true
47
- shift_range: [-0.3, 0.3]
48
- apply_background: false
49
- background_range: [1.0e-10, 1.0e-4]
50
- add_to_context: true
51
-
52
- # q_noise:
53
- # cls: BasicQNoiseGenerator
54
- # kwargs:
55
- # apply_systematic_shifts: true
56
- # shift_std: 1.0e-3
57
- # apply_gaussian_noise: false
58
- # noise_std: [0., 1.0e-3]
59
- # add_to_context: true
60
-
61
- smearing:
62
- cls: Smearing
63
- kwargs:
64
- sigma_range: [0.01, 0.08]
65
- constant_dq: False
66
- gauss_num: 17
67
- share_smeared: 0.8
68
-
69
- curves_scaler:
70
- cls: LogAffineCurvesScaler
71
- kwargs:
72
- weight: 0.2
73
- bias: 1.0
74
- eps: 1.0e-10
75
-
76
- model:
77
- network:
78
- cls: NetworkWithPriors
79
- pretrained_name: null
80
- device: 'cuda'
81
- kwargs:
82
- embedding_net_type: 'conv'
83
- embedding_net_kwargs:
84
- in_channels: 1
85
- hidden_channels: [32, 64, 128, 256, 512]
86
- kernel_size: 3
87
- dim_embedding: 128
88
- dim_avpool: 1
89
- use_batch_norm: true
90
- use_se: false
91
- activation: 'gelu'
92
- pretrained_embedding_net: null
93
- dim_out: 8
94
- dim_conditioning_params: 0
95
- layer_width: 512
96
- num_blocks: 6
97
- repeats_per_block: 2
98
- residual: true
99
- use_batch_norm: true
100
- use_layer_norm: false
101
- mlp_activation: 'gelu'
102
- dropout_rate: 0.0
103
- tanh_output: false
104
- conditioning: 'concat' # 'concat', 'glu', 'film'
105
- concat_condition_first_layer: false
106
-
107
- training:
108
- trainer_cls: PointEstimatorTrainer
109
- num_iterations: 10000
110
- batch_size: 4096
111
- lr: 1.0e-3
112
- grad_accumulation_steps: 1
113
- clip_grad_norm_max: null
114
- update_tqdm_freq: 1
115
- optimizer: AdamW
116
- trainer_kwargs:
117
- train_with_q_input: false
118
- train_with_sigmas: false
119
- condition_on_q_resolutions: false
120
- use_curve_reconstruction_loss: false
121
- rescale_loss_interval_width: true #false
122
- use_l1_loss: true #false
123
- optim_kwargs:
124
- betas: [0.9, 0.999]
125
- weight_decay: 0.0005
126
- callbacks:
127
- save_best_model:
128
- enable: true
129
- freq: 500
130
- lr_scheduler:
131
- cls: CosineAnnealingWithWarmup
132
- kwargs:
133
- min_lr: 1.0e-6
134
- warmup_iters: 100
135
- total_iters: 10000
136
- # cls: StepLR
137
- # kwargs:
138
- # step_size: 10000
139
- # gamma: 0.5
140
-
141
- # logger:
142
- # cls: TensorBoardLogger
143
- # kwargs:
144
- # log_dir: "tensorboard_runs/test_1"