correct_for_frame_skip¶
- torchrl.trainers.helpers.correct_for_frame_skip(cfg: DictConfig) DictConfig [來源]¶
修正輸入 frame_skip 的引數,方法是將所有反映 frame 計數的引數除以 frame_skip。
這樣做是為了避免在不知情的情況下從環境中過度取樣,即目標是總幀數為 1M,但實際上收集了 frame_skip * 1M 幀。
- 參數:
cfg (DictConfig) – 包含一些 frame 計數引數的 DictConfig,包括:「max_frames_per_traj」、「total_frames」、「frames_per_batch」、「record_frames」、「annealing_frames」、「init_random_frames」、「init_env_steps」
- 傳回:
輸入的 DictConfig,就地修改。