add stop worker

45ec57e9 · tangwei · 3045ed26 · 45ec57e9
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

fleetrec/core/trainers/ctr_coding_trainer.py fleetrec/core/trainers/ctr_coding_trainer.py +1 -0

未找到文件。
--- a/fleetrec/core/trainers/ctr_coding_trainer.py
+++ b/fleetrec/core/trainers/ctr_coding_trainer.py
@@ -104,6 +104,7 @@ class CtrPaddleTrainer(Trainer):

    def server(self, context):
        fleet.run_server()
+        fleet.stop_worker()
        context['is_exit'] = True

    def train(self, context):