diff --git a/fleetrec/core/trainers/ctr_coding_trainer.py b/fleetrec/core/trainers/ctr_coding_trainer.py index 34f8fa44e8eecb960bb01445397f0184c56cd3e5..1d8429be0eddbba83abc686be710434085a1a804 100755 --- 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):