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) – 环境创建器的关键字参数
