Method__init__(
self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation="gelu"
)
athena/layers/u2/transformer_u2_layer.py:160
Method__init__(
self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation="gelu"
)
athena/layers/u2/conformer_u2_layer.py:178