fix predict bug

2021-11-06 14:31:56 +08:00 · 2021-11-06 14:31:56 +08:00 · 0120000e91
parent f7efd3fc87
commit 0120000e91
1 changed files with 0 additions and 2 deletions
--- a/src/deepke/name_entity_re/few_shot/module/train.py
+++ b/src/deepke/name_entity_re/few_shot/module/train.py
@ -109,8 +109,6 @@ class Trainer(object):
        self.model.train()

    def predict(self):
-        assert self.load_path is not None and self.test_data is not None
-
        self.model.eval()
        self.logger.info("***** Running testing *****")
        self.logger.info("  Num instance = %d", len(self.test_data)*self.batch_size)