Skip to content

Commit 9287193

Browse files
committed
adjust mambular basemodel to new hparams
1 parent 4da578d commit 9287193

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

mambular/base_models/mambular.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,10 @@ def __init__(
105105
dt_init_floor=self.hparams.get("dt_init_floor", config.dt_init_floor),
106106
norm=globals()[self.hparams.get("norm", config.norm)],
107107
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+
),
108112
)
109113

110114
norm_layer = self.hparams.get("norm", config.norm)

0 commit comments

Comments
 (0)