ab_upt.yamlΒΆ

recipes/aero_cfd/configs/model/ab_upt.yaml

 1kind: noether.modeling.models.AeroABUPT
 2name: ab_upt
 3geometry_depth: 1
 4hidden_dim: 192
 5transformer_block_config:
 6  num_heads: 3
 7  mlp_expansion_factor: 4
 8  use_rope: true
 9supernode_pooling_config:
10  input_dim: ${data_specs.position_dim}
11  radius: 1
12physics_blocks: 
13  - perceiver
14  - self
15  - cross
16  - self
17  - cross
18  - self
19  - cross
20  - self
21  - cross
22  - self
23num_domain_decoder_blocks:
24  surface: 2
25  volume: 2
26optimizer_config: ${optimizer}
27forward_properties:
28  - geometry_position
29  - geometry_supernode_idx
30  - geometry_batch_idx
31  - surface_anchor_position
32  - volume_anchor_position
33data_specs: ${data_specs}