p71924506
/
deepke
1
0
Fork 0
Code Issues Pull Requests 1 Projects Releases Wiki Activity

fix predict bug

This commit is contained in:
lilei 2021-11-06 14:31:56 +08:00 committed by GitHub
parent f7efd3fc87
commit 0120000e91
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/deepke/name_entity_re/few_shot/module/train.py
View File

@ -109,8 +109,6 @@ class Trainer(object):
self.model.train() self.model.train()
def predict(self): def predict(self):
assert self.load_path is not None and self.test_data is not None
self.model.eval() self.model.eval()
self.logger.info("***** Running testing *****") self.logger.info("***** Running testing *****")
self.logger.info(" Num instance = %d", len(self.test_data)*self.batch_size) self.logger.info(" Num instance = %d", len(self.test_data)*self.batch_size)