From f5dc2a651c91ac403d58b494f4dfd9dd3b677d41 Mon Sep 17 00:00:00 2001
From: wuzewu <wuzewu@baidu.com>
Date: Thu, 23 Apr 2020 20:39:15 +0800
Subject: [PATCH] update nlp_module

---
 paddlehub/module/module.py     | 4 ++--
 paddlehub/module/nlp_module.py | 7 +++++++
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/paddlehub/module/module.py b/paddlehub/module/module.py
index b73c5533..39d7e63f 100644
--- a/paddlehub/module/module.py
+++ b/paddlehub/module/module.py
@@ -257,8 +257,8 @@ class Module(fluid.dygraph.Layer):
     def _initialize(self):
         pass
 
-    def forward(self, *args):
-        return self.model_runner(*args)
+    def forward(self, *args, **kwargs):
+        return self.model_runner(*args, **kwargs)
 
 
 class ModuleHelper(object):
diff --git a/paddlehub/module/nlp_module.py b/paddlehub/module/nlp_module.py
index f99a672d..c7b67371 100644
--- a/paddlehub/module/nlp_module.py
+++ b/paddlehub/module/nlp_module.py
@@ -353,6 +353,13 @@ class TransformerModule(NLPBaseModule):
 
         return inputs, outputs, module_program
 
+    @property
+    def model_runner(self):
+        if not self._model_runner:
+            self._model_runner = fluid.dygraph.StaticModelRunner(
+                self.params_path)
+        return self._model_runner
+
     def get_embedding(self, texts, use_gpu=False, batch_size=1):
         """
         get pooled_output and sequence_output for input texts.
-- 
GitLab