ray/rllib/algorithms/r2d2/__init__.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

12 lines
334 B
Python
Raw Normal View History

from ray.rllib.algorithms.r2d2.r2d2 import R2D2, R2D2Config, R2D2_DEFAULT_CONFIG
from ray.rllib.algorithms.r2d2.r2d2_tf_policy import R2D2TFPolicy
from ray.rllib.algorithms.r2d2.r2d2_torch_policy import R2D2TorchPolicy
__all__ = [
"R2D2",
"R2D2Config",
"R2D2TFPolicy",
"R2D2TorchPolicy",
"R2D2_DEFAULT_CONFIG",
]