transformer.yamlΒΆ

recipes/aero_cfd/configs/model/transformer.yaml

 1kind: noether.modeling.models.AeroTransformer
 2name: transformer
 3hidden_dim: 192
 4transformer_block_config: 
 5  num_heads: 3
 6  mlp_expansion_factor: 4
 7  use_rope: true
 8depth: 12
 9optimizer_config: ${optimizer}
10data_specs: ${data_specs}
11forward_properties:
12  - surface_position
13  - volume_position
14  - surface_features
15  - volume_features