make_collector_onpolicy¶
- torchrl.trainers.helpers.make_collector_onpolicy(make_env: Callable[[], EnvBase], actor_model_explore: TensorDictModuleWrapper | ProbabilisticTensorDictSequential, cfg: DictConfig, make_env_kwargs: dict | None = None) DataCollectorBase[源代码]¶
在同策设置中创建一个 collector。
- 参数:
make_env (Callable) – environment creator
actor_model_explore (SafeModule) – 用于评估和探索更新的模型实例
cfg (DictConfig) – 创建 collector 对象的配置
make_env_kwargs (dict) – env creator 的 kwargs
