↓ 1 callersMethod__init__(
self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation="gelu",
unidirec
athena/layers/u2/transformer_u2_layer.py:45
↓ 1 callersMethod__init__(
self, d_model, nhead, cnn_module_kernel=15, dim_feedforward=2048, dropout=0.1, activation="gelu"
athena/layers/u2/conformer_u2_layer.py:49