multi_gpu_impl.py 397 B

1234567891011
  1. from ray.rllib.policy.dynamic_tf_policy import TFMultiGPUTowerStack
  2. from ray.rllib.utils.deprecation import deprecation_warning
  3. # Backward compatibility.
  4. deprecation_warning(
  5. old="ray.rllib.execution.multi_gpu_impl.LocalSyncParallelOptimizer",
  6. new="ray.rllib.policy.dynamic_tf_policy.TFMultiGPUTowerStack",
  7. error=False,
  8. )
  9. # Old name.
  10. LocalSyncParallelOptimizer = TFMultiGPUTowerStack