| # Variational Autoencoder for compression-only shells | |
| # Extension of Pastrana et al. (ICLR 2025), Section 6.1 | |
| # | |
| # Identical to bezier.yml except: | |
| # - loss.vae section with beta annealing parameters | |
| # - Used with model name "variational_formfinder" | |
| seed: 91 | |
| generator: | |
| name: "bezier_symmetric_double" | |
| bounds: "saddle" | |
| num_uv: 10 | |
| size: 10.0 | |
| num_points: 4 | |
| lerp_factor: 0.5 | |
| fdm: | |
| load: -0.5 | |
| encoder: | |
| shift: 0.0 | |
| hidden_layer_size: 256 | |
| hidden_layer_num: 3 | |
| activation_fn_name: "elu" | |
| final_activation_fn_name: "softplus" | |
| decoder: | |
| include_params_xl: True | |
| hidden_layer_size: 256 | |
| hidden_layer_num: 3 | |
| activation_fn_name: "elu" | |
| loss: | |
| shape: | |
| include: True | |
| weight: 1.0 | |
| residual: | |
| include: True | |
| weight: 1.0 | |
| vae: | |
| # Cyclical beta annealing (Fu et al., NAACL 2019) | |
| beta_max: 1.0 # Maximum KL weight (1.0 = standard ELBO) | |
| cycle_length: 5000 # Steps per annealing cycle | |
| warmup_ratio: 0.5 # Fraction of cycle for linear warmup | |
| optimizer: | |
| name: "adam" | |
| learning_rate: 3.0e-5 | |
| clip_norm: 0.0 | |
| training: | |
| steps: 10000 | |
| batch_size: 64 | |