ray/rllib/algorithms/sac/__init__.py

21 lines
608 B
Python

from ray.rllib.algorithms.sac.sac import SACTrainer, DEFAULT_CONFIG, SACConfig
from ray.rllib.algorithms.sac.sac_tf_policy import SACTFPolicy
from ray.rllib.algorithms.sac.sac_torch_policy import SACTorchPolicy
from ray.rllib.algorithms.sac.rnnsac import (
RNNSACTrainer,
DEFAULT_CONFIG as RNNSAC_DEFAULT_CONFIG,
)
from ray.rllib.algorithms.sac.rnnsac import RNNSACTorchPolicy, RNNSACConfig
__all__ = [
"DEFAULT_CONFIG",
"SACTFPolicy",
"SACTorchPolicy",
"SACTrainer",
"SACConfig",
"RNNSAC_DEFAULT_CONFIG",
"RNNSACTorchPolicy",
"RNNSACTrainer",
"RNNSACConfig",
]