add kwargs

HessTaha · May 31, 2020 · d03fe60 · d03fe60
1 parent c554f2e
commit d03fe60
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/model.py b/src/model.py
@@ -24,7 +24,8 @@ def __init__(self,
                 ffn: int = 128, 
                 n_classes: int = None,
                 do_normalization: bool = True,
-                pooling_strategy: str = 'avg'):
+                pooling_strategy: str = 'avg',
+                **kwargs):
 
         super(LSTMModel, self).__init__()
         self.embedding = torch.nn.Embedding(vocab_size, embed_dim)