mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-09-27 04:46:16 +08:00
fix rl bugs (#3654)
This commit is contained in:
@@ -1158,10 +1158,6 @@ class FDConfig:
|
||||
"""
|
||||
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]
|
||||
|
||||
if self.parallel_config.tensor_parallel_size <= self.worker_num_per_node:
|
||||
|
Reference in New Issue
Block a user