mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-09-27 12:52:29 +08:00
fix rl bugs (#3654)
This commit is contained in:
@@ -1158,10 +1158,6 @@ class FDConfig:
|
|||||||
"""
|
"""
|
||||||
calculate some parameters
|
calculate some parameters
|
||||||
"""
|
"""
|
||||||
assert (
|
|
||||||
self.device_ids.split(",").__len__() == self.worker_num_per_node
|
|
||||||
), f"invalid CUDA_VISIBLE_DEVICES, should be equal to {self.worker_num_per_node}"
|
|
||||||
|
|
||||||
self.local_device_ids = self.device_ids.split(",")[: self.parallel_config.tensor_parallel_size]
|
self.local_device_ids = self.device_ids.split(",")[: self.parallel_config.tensor_parallel_size]
|
||||||
|
|
||||||
if self.parallel_config.tensor_parallel_size <= self.worker_num_per_node:
|
if self.parallel_config.tensor_parallel_size <= self.worker_num_per_node:
|
||||||
|
Reference in New Issue
Block a user