diff --git a/pypots/imputation/informer/modules/core.py b/pypots/imputation/informer/modules/core.py index 306b693a..455a7b1a 100644 --- a/pypots/imputation/informer/modules/core.py +++ b/pypots/imputation/informer/modules/core.py @@ -33,7 +33,7 @@ def __init__( self.seq_len = n_steps self.n_layers = n_layers self.enc_embedding = DataEmbedding( - n_features * 2, + n_features, d_model, dropout=dropout, )