mirror of
https://github.com/vale981/ray
synced 2025-03-07 02:51:39 -05:00
26 lines
934 B
Python
26 lines
934 B
Python
from ray.rllib.agents.dqn.apex import ApexTrainer
|
|
from ray.rllib.agents.dqn.dqn import DQNTrainer, DEFAULT_CONFIG
|
|
from ray.rllib.agents.dqn.dqn_tf_policy import DQNTFPolicy
|
|
from ray.rllib.agents.dqn.dqn_torch_policy import DQNTorchPolicy
|
|
from ray.rllib.agents.dqn.r2d2 import R2D2Trainer, DEFAULT_CONFIG as \
|
|
R2D2_DEFAULT_CONFIG
|
|
from ray.rllib.agents.dqn.r2d2_torch_policy import R2D2TorchPolicy
|
|
from ray.rllib.agents.dqn.simple_q import SimpleQTrainer, \
|
|
DEFAULT_CONFIG as SIMPLE_Q_DEFAULT_CONFIG
|
|
from ray.rllib.agents.dqn.simple_q_tf_policy import SimpleQTFPolicy
|
|
from ray.rllib.agents.dqn.simple_q_torch_policy import SimpleQTorchPolicy
|
|
|
|
__all__ = [
|
|
"ApexTrainer",
|
|
"DQNTFPolicy",
|
|
"DQNTorchPolicy",
|
|
"DQNTrainer",
|
|
"DEFAULT_CONFIG",
|
|
"R2D2TorchPolicy",
|
|
"R2D2Trainer",
|
|
"R2D2_DEFAULT_CONFIG",
|
|
"SIMPLE_Q_DEFAULT_CONFIG",
|
|
"SimpleQTFPolicy",
|
|
"SimpleQTorchPolicy",
|
|
"SimpleQTrainer",
|
|
]
|