Efradeca commited on
Commit
535db88
·
verified ·
1 Parent(s): 5d28aab

Upload scripts/variational_bezier.yml with huggingface_hub

Browse files
Files changed (1) hide show
  1. scripts/variational_bezier.yml +54 -0
scripts/variational_bezier.yml ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Variational Autoencoder for compression-only shells
2
+ # Extension of Pastrana et al. (ICLR 2025), Section 6.1
3
+ #
4
+ # Identical to bezier.yml except:
5
+ # - loss.vae section with beta annealing parameters
6
+ # - Used with model name "variational_formfinder"
7
+
8
+ seed: 91
9
+
10
+ generator:
11
+ name: "bezier_symmetric_double"
12
+ bounds: "saddle"
13
+ num_uv: 10
14
+ size: 10.0
15
+ num_points: 4
16
+ lerp_factor: 0.5
17
+
18
+ fdm:
19
+ load: -0.5
20
+
21
+ encoder:
22
+ shift: 0.0
23
+ hidden_layer_size: 256
24
+ hidden_layer_num: 3
25
+ activation_fn_name: "elu"
26
+ final_activation_fn_name: "softplus"
27
+
28
+ decoder:
29
+ include_params_xl: True
30
+ hidden_layer_size: 256
31
+ hidden_layer_num: 3
32
+ activation_fn_name: "elu"
33
+
34
+ loss:
35
+ shape:
36
+ include: True
37
+ weight: 1.0
38
+ residual:
39
+ include: True
40
+ weight: 1.0
41
+ vae:
42
+ # Cyclical beta annealing (Fu et al., NAACL 2019)
43
+ beta_max: 1.0 # Maximum KL weight (1.0 = standard ELBO)
44
+ cycle_length: 5000 # Steps per annealing cycle
45
+ warmup_ratio: 0.5 # Fraction of cycle for linear warmup
46
+
47
+ optimizer:
48
+ name: "adam"
49
+ learning_rate: 3.0e-5
50
+ clip_norm: 0.0
51
+
52
+ training:
53
+ steps: 10000
54
+ batch_size: 64