ray/rllib/execution/multi_gpu_learner.py

14 lines
439 B
Python
Raw Normal View History

from ray.rllib.execution.multi_gpu_learner_thread import \
MultiGPULearnerThread, _MultiGPULoaderThread
from ray.rllib.utils.deprecation import deprecation_warning
# Backward compatibility.
deprecation_warning(
old="ray.rllib.execution.multi_gpu_learner.py",
new="ray.rllib.execution.multi_gpu_learner_thread.py",
error=False,
)
# Old names.
TFMultiGPULearner = MultiGPULearnerThread
_LoaderThread = _MultiGPULoaderThread