DataCollectorBase

class torchrl.collectors.DataCollectorBase

資料收集器的基底類別。

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

如果資料收集器的策略和經過訓練的策略位於不同的裝置上，則更新策略權重。

參數:

policy_weights (TensorDictBase, optional) – 如果提供，則為包含用於更新的策略權重的 TensorDict。