multi_gpu_learner.py 439 B

12345678910111213
  1. from ray.rllib.execution.multi_gpu_learner_thread import \
  2. MultiGPULearnerThread, _MultiGPULoaderThread
  3. from ray.rllib.utils.deprecation import deprecation_warning
  4. # Backward compatibility.
  5. deprecation_warning(
  6. old="ray.rllib.execution.multi_gpu_learner.py",
  7. new="ray.rllib.execution.multi_gpu_learner_thread.py",
  8. error=False,
  9. )
  10. # Old names.
  11. TFMultiGPULearner = MultiGPULearnerThread
  12. _LoaderThread = _MultiGPULoaderThread