2020-09-21 23:03:06 -07:00
|
|
|
from ray.rllib.execution.concurrency_ops import Concurrently, Enqueue, Dequeue
|
2021-07-20 14:58:13 -04:00
|
|
|
from ray.rllib.execution.learner_thread import LearnerThread
|
2020-09-21 23:03:06 -07:00
|
|
|
from ray.rllib.execution.metric_ops import StandardMetricsReporting, \
|
|
|
|
|
CollectMetrics, OncePerTimeInterval, OncePerTimestepsElapsed
|
2021-07-20 14:58:13 -04:00
|
|
|
from ray.rllib.execution.multi_gpu_learner_thread import MultiGPULearnerThread
|
2021-11-19 11:57:37 +01:00
|
|
|
from ray.rllib.execution.buffers.replay_buffer import ReplayBuffer, \
|
2020-09-21 23:03:06 -07:00
|
|
|
PrioritizedReplayBuffer
|
2021-11-19 11:57:37 +01:00
|
|
|
from ray.rllib.execution.buffers.multi_agent_replay_buffer import \
|
|
|
|
|
MultiAgentReplayBuffer
|
2020-09-21 23:03:06 -07:00
|
|
|
from ray.rllib.execution.replay_ops import StoreToReplayBuffer, Replay, \
|
|
|
|
|
SimpleReplayBuffer, MixInReplay
|
|
|
|
|
from ray.rllib.execution.rollout_ops import ParallelRollouts, AsyncGradients, \
|
|
|
|
|
ConcatBatches, SelectExperiences, StandardizeFields
|
2021-07-20 14:58:13 -04:00
|
|
|
from ray.rllib.execution.train_ops import TrainOneStep, MultiGPUTrainOneStep, \
|
2020-09-21 23:03:06 -07:00
|
|
|
ComputeGradients, ApplyGradients, AverageGradients, UpdateTargetNetwork
|
|
|
|
|
|
|
|
|
|
__all__ = [
|
|
|
|
|
"ApplyGradients",
|
|
|
|
|
"AsyncGradients",
|
|
|
|
|
"AverageGradients",
|
|
|
|
|
"CollectMetrics",
|
|
|
|
|
"ComputeGradients",
|
|
|
|
|
"ConcatBatches",
|
|
|
|
|
"Concurrently",
|
|
|
|
|
"Dequeue",
|
|
|
|
|
"Enqueue",
|
2021-07-20 14:58:13 -04:00
|
|
|
"LearnerThread",
|
2020-09-21 23:03:06 -07:00
|
|
|
"MixInReplay",
|
2021-11-19 11:57:37 +01:00
|
|
|
"MultiAgentReplayBuffer",
|
2021-07-20 14:58:13 -04:00
|
|
|
"MultiGPULearnerThread",
|
2020-09-21 23:03:06 -07:00
|
|
|
"OncePerTimeInterval",
|
|
|
|
|
"OncePerTimestepsElapsed",
|
|
|
|
|
"ParallelRollouts",
|
|
|
|
|
"PrioritizedReplayBuffer",
|
|
|
|
|
"Replay",
|
|
|
|
|
"ReplayBuffer",
|
|
|
|
|
"SelectExperiences",
|
|
|
|
|
"SimpleReplayBuffer",
|
|
|
|
|
"StandardMetricsReporting",
|
|
|
|
|
"StandardizeFields",
|
|
|
|
|
"StoreToReplayBuffer",
|
|
|
|
|
"TrainOneStep",
|
2021-07-20 14:58:13 -04:00
|
|
|
"MultiGPUTrainOneStep",
|
2020-09-21 23:03:06 -07:00
|
|
|
"UpdateTargetNetwork",
|
|
|
|
|
]
|
