apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 13:28:13 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

update rl signal (#5650)

Browse Source
This commit is contained in:
Yuanle Liu
2025-12-19 12:04:18 +08:00
committed by GitHub
parent 23bfd28624
commit b3f78815d8
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
fastdeploy/worker/worker_process.py
View File

@@ -427,7 +427,8 @@ class PaddleDisWorkerProc:
if self.fd_config.load_config.dynamic_load_weight:
if self.model_weights_status.value[0] != ModelWeightsStatus.NORMAL:
self.model_weights_signal[0] = int(self.model_weights_status.value[0])
self.model_weights_signal[0] = self._broadcast_model_weights_signal(src=0, group=None)
if self.ranks > 1:
self.model_weights_signal[0] = self._broadcast_model_weights_signal(src=0, group=None)
self.insert_step = False
req_dicts = None
@@ -450,7 +451,8 @@ class PaddleDisWorkerProc:
if self.fd_config.load_config.dynamic_load_weight:
if self.model_weights_signal[0] != ModelWeightsStatus.NORMAL:
paddle.distributed.barrier()
if self.ranks > 1:
paddle.distributed.barrier()
logger.info(
f"Rank: {self.local_rank} to update or clear parameters, signal is {self.model_weights_signal[0]}, [-1:clear, 1:update]"
)