Browse Source

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

tags/v1.0.0alpha
WillQvQ 2 years ago
parent
commit
2e3a689b29
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      fastNLP/core/tester.py

+ 1
- 1
fastNLP/core/tester.py View File

@@ -148,7 +148,7 @@ class Tester(object):
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:


Loading…
Cancel
Save