File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -47,7 +47,7 @@ class WeightUpdaterBase(metaclass=abc.ABCMeta):
47
47
48
48
Attributes:
49
49
collector: The collector (or any container) of the weight receiver. The collector is registered via
50
- :meth:`~torchrl.collectors.WeightUpdateReceiverBase .register_collector`.
50
+ :meth:`~torchrl.collectors.WeightUpdaterBase .register_collector`.
51
51
52
52
Methods:
53
53
push_weights: Updates the weights on specified or all remote workers.
@@ -317,7 +317,7 @@ class VanillaWeightUpdater(WeightUpdaterBase):
317
317
This class is used by default in the `SyncDataCollector` when no custom weight sender
318
318
is provided.
319
319
320
- .. seealso:: :class:`~torchrl.collectors.WeightUpdateReceiverBase ` and :class:`~torchrl.collectors.SyncDataCollector`.
320
+ .. seealso:: :class:`~torchrl.collectors.WeightUpdaterBase ` and :class:`~torchrl.collectors.SyncDataCollector`.
321
321
322
322
Keyword Args:
323
323
weight_getter (Callable[[], TensorDictBase], optional): a callable that returns the weights from the server.
You can’t perform that action at this time.
0 commit comments