From d03fe6074e1eb23c409e60011ffe1bf818c4ddbd Mon Sep 17 00:00:00 2001
From: HessTaha <tahahessane2@gmail.com>
Date: Sun, 31 May 2020 17:13:46 +0200
Subject: [PATCH] add kwargs

---
 src/model.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/model.py b/src/model.py
index a199e1d..a384187 100644
--- 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)