hummingbird
/
fastNLP
Not watched
2
0
Fork 0
Code
Releases 13 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

修复Tester初始化bug (#401) 

Co-authored-by: ouyhlan <ouyhlan@icloud.com>
dev^2
Ou GitHub 3 years ago
parent
9ac7d09431
commit
a4f792d0cc
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  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:


Write Preview
Loading…
Cancel
Save