We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4da578d commit 9287193Copy full SHA for 9287193
1 file changed
mambular/base_models/mambular.py
@@ -105,6 +105,10 @@ def __init__(
105
dt_init_floor=self.hparams.get("dt_init_floor", config.dt_init_floor),
106
norm=globals()[self.hparams.get("norm", config.norm)],
107
activation=self.hparams.get("activation", config.activation),
108
+ bidirectional=self.hparams.get("bidiretional", config.bidirectional),
109
+ use_learnable_interaction=self.hparams.get(
110
+ "use_learnable_interactions", config.use_learnable_interaction
111
+ ),
112
)
113
114
norm_layer = self.hparams.get("norm", config.norm)
0 commit comments