修复训练时config epoch不起效果的问题

wuyunxiangwyx · Oct 22, 2020 · 6d9c009 · 6d9c009
1 parent 33078dd
commit 6d9c009
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ltp/exe/executor.py b/ltp/exe/executor.py
@@ -39,7 +39,7 @@ def __init__(self, config: Config):
         self.tasks = self.config.tasks
         self.epoch_size = self.config.epoch_size
 
-    def train(self, epoch: int = 30):
+    def train(self, epoch: int = None):
         epoch = self.config.epoch if epoch is None else epoch
 
         # =================================== 任务初始化 =================================