ray/rllib/execution/multi_gpu_impl.py

11 lines
397 B
Python

from ray.rllib.policy.dynamic_tf_policy import TFMultiGPUTowerStack
from ray.rllib.utils.deprecation import deprecation_warning
# Backward compatibility.
deprecation_warning(
old="ray.rllib.execution.multi_gpu_impl.LocalSyncParallelOptimizer",
new="ray.rllib.policy.dynamic_tf_policy.TFMultiGPUTowerStack",
error=False,
)
# Old name.
LocalSyncParallelOptimizer = TFMultiGPUTowerStack