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
