fix from_config

lhbonifacio · Jan 19, 2019 · e452d21 · e452d21
1 parent 731d39b
commit e452d21
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/modules/layers/decoders.py b/modules/layers/decoders.py
@@ -661,15 +661,15 @@ def create(cls, label_size, input_dim, input_dropout=0.5, key_dim=64,
 
 class NCRFDecoder(nn.Module):
 
-    @property
+    # TODO: TRY TO FIX THIS SHIT (get attribute error)
+    # @property WTF!!!
     def config(self):
         config = {
             "name": "NCRFDecoder",
             "params": {
                 "label_size": self.label_size,
                 "input_dim": self.input_dim,
                 "input_dropout": self.dropout.p,
-                "use_cuda": self.use_cuda,
                 "nbest": self.nbest
             }
         }

diff --git a/modules/models/bert_models.py b/modules/models/bert_models.py
@@ -15,7 +15,7 @@ def config(self):
                 "name": self.__class__.__name__,
                 "params": {
                     "encoder": self.encoder.config,
-                    "decoder": self.decoder.config,
+                    "decoder": self.decoder.config(),
                     "use_cuda": self.use_cuda
                 }
             }