ray.rllib.algorithms.algorithm_config.AlgorithmConfig.get_rollout_片段_长度#

AlgorithmConfig.get_rollout_fragment_length(worker_index: int = 0) int[源代码]#

如果设置为“auto”,则会自动推断出适当的 rollout_fragment_length 设置。

使用简单的公式:rollout_fragment_length = total_train_batch_size / (num_envs_per_env_runner * num_env_runners)

如果结果是一个分数并且提供了 worker_index,将会使那些工作线程增加额外的步数,从而使得整个批处理大小(跨工作线程)将精确地增加到 total_train_batch_size

返回:

用户提供的 rollout_fragment_length 或一个计算得到的值(如果用户提供的值是“auto”),确保在每次迭代中 total_train_batch_size 精确达到。