correct_for_frame_skip¶
- torchrl.trainers.helpers.correct_for_frame_skip(cfg: DictConfig) DictConfig [源代码]¶
修正输入 frame_skip 的参数,通过将反映帧数的参数全部除以 frame_skip 来实现。
这样可以避免无意中过度采样环境,即目标是收集 100 万帧,但实际上收集了 frame_skip * 100 万帧。
- 参数:
cfg (DictConfig) – 包含一些帧计数参数的 DictConfig,包括:“max_frames_per_traj”, “total_frames”, “frames_per_batch”, “record_frames”, “annealing_frames”, “init_random_frames”, “init_env_steps”
- 返回:
输入的 DictConfig,原地修改。