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
精确达到。