Flexpert-Design/src/models/configs/ESMIF.yaml

augment_eps: 0.0
num_encoder_layers: 4
hidden_dim: 128
top_k_neighbors: 30
encoder_embed_dim: 512
decoder_embed_dim: 512
dropout: 0.1
node_hidden_dim_scalar: 1024
node_hidden_dim_vector: 256
edge_hidden_dim_scalar: 32
edge_hidden_dim_vector: 1
encoder_layers: 8
encoder_attention_heads: 8
attention_dropout: 0.1
encoder_ffn_embed_dim: 2048
decoder_layers: 8
decoder_attention_heads: 8
decoder_ffn_embed_dim: 2048