mirror of
https://github.com/vale981/ray
synced 2025-03-06 02:21:39 -05:00
11 lines
334 B
Python
11 lines
334 B
Python
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",
|
|
]
|