12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849 |
- from ray.rllib.execution.concurrency_ops import Concurrently, Enqueue, Dequeue
- from ray.rllib.execution.learner_thread import LearnerThread
- from ray.rllib.execution.metric_ops import StandardMetricsReporting, \
- CollectMetrics, OncePerTimeInterval, OncePerTimestepsElapsed
- from ray.rllib.execution.multi_gpu_learner_thread import MultiGPULearnerThread
- from ray.rllib.execution.buffers.replay_buffer import ReplayBuffer, \
- PrioritizedReplayBuffer
- from ray.rllib.execution.buffers.multi_agent_replay_buffer import \
- MultiAgentReplayBuffer
- from ray.rllib.execution.replay_ops import StoreToReplayBuffer, Replay, \
- SimpleReplayBuffer, MixInReplay
- from ray.rllib.execution.rollout_ops import ParallelRollouts, AsyncGradients, \
- ConcatBatches, SelectExperiences, StandardizeFields, \
- synchronous_parallel_sample
- from ray.rllib.execution.train_ops import TrainOneStep, MultiGPUTrainOneStep, \
- ComputeGradients, ApplyGradients, AverageGradients, UpdateTargetNetwork, \
- train_one_step
- __all__ = [
- "synchronous_parallel_sample",
- "train_one_step",
- "ApplyGradients",
- "AsyncGradients",
- "AverageGradients",
- "CollectMetrics",
- "ComputeGradients",
- "ConcatBatches",
- "Concurrently",
- "Dequeue",
- "Enqueue",
- "LearnerThread",
- "MixInReplay",
- "MultiAgentReplayBuffer",
- "MultiGPULearnerThread",
- "OncePerTimeInterval",
- "OncePerTimestepsElapsed",
- "ParallelRollouts",
- "PrioritizedReplayBuffer",
- "Replay",
- "ReplayBuffer",
- "SelectExperiences",
- "SimpleReplayBuffer",
- "StandardMetricsReporting",
- "StandardizeFields",
- "StoreToReplayBuffer",
- "TrainOneStep",
- "MultiGPUTrainOneStep",
- "UpdateTargetNetwork",
- ]
|