fix transformer yaml

Philip Janz · Philip Janz · commit c7905452414e · 2025-12-15T16:29:19.000+01:00
diff --git a/cybench/conf/model/torch_model/temporal_encoder/processor/transformer.yaml b/cybench/conf/model/torch_model/temporal_encoder/processor/transformer.yaml
@@ -4,7 +4,7 @@ embed_dim: ${model.torch_model.embed_dim}
 
 num_layers: 1       # number of TransformerEncoder layers
 nhead: 4             # attention heads
-dim_feedforward: 128 # FFN hidden size
+dim_feedforward: ${model.torch_model.embed_dim} # FFN hidden size
 dropout: 0.2
 
 _search_:
