修复Tester初始化bug (#401)

Co-authored-by: ouyhlan <[email protected]>
fastnlp · Dec 2, 2021 · a4f792d · a4f792d
1 parent 9ac7d09
commit a4f792d
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: