ray/rllib/execution/multi_gpu_learner.py

15 lines
446 B
Python

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