ab_upt.yamlΒΆ

recipes/heat_transfer/configs/model/ab_upt.yaml

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