Merge branch 'dev' of github.com:fastNLP/fastNLP into dev

fastnlp · Dec 6, 2021 · 2e3a689 · 2e3a689
2 parents ca6c9dd + a4f792d
commit 2e3a689
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fastNLP/core/tester.py b/fastNLP/core/tester.py
@@ -148,7 +148,7 @@ def __init__(self, data, model, metrics, batch_size=16, num_workers=0, device=No
                 self._predict_func = self._model.predict
                 self._predict_func_wrapper = self._model.predict
         else:
-            if _model_contains_inner_module(model):
+            if _model_contains_inner_module(self._model):
                 self._predict_func_wrapper = self._model.forward
                 self._predict_func = self._model.module.forward
             else: