make_collector_onpolicy¶
- torchrl.trainers.helpers.make_collector_onpolicy(make_env: Callable[], EnvBase], actor_model_explore: Union[TensorDictModuleWrapper, ProbabilisticTensorDictSequential], cfg: DictConfig, make_env_kwargs: Optional[Dict] = None) DataCollectorBase[原始碼]¶
在 on-policy（同策略）設定中建立一個收集器。
- 引數:
make_env (Callable) – 環境建立器
actor_model_explore (SafeModule) – 用於評估和探索更新的模型例項
cfg (DictConfig) – 用於建立收集器物件的配置
make_env_kwargs (dict) – 環境建立器的 kwargs（關鍵字引數）
