diff --git a/src/deepke/name_entity_re/few_shot/module/train.py b/src/deepke/name_entity_re/few_shot/module/train.py
index 00bcb1a..0d4ff0e 100644
--- 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)