DataCollectorBase

class torchrl.collectors.DataCollectorBase(*args, **kwds)

数据收集器的基类。

update_policy_weights_(policy_weights: Optional[TensorDictBase] = None) → None

如果数据收集器的策略和训练策略位于不同的设备上，则更新策略权重。

参数：

policy_weights (TensorDictBase, 可选) – 如果提供，则为包含要用于更新的策略权重的 TensorDict。