reference/generated/torchrl.weight_update.InitialSyncTransport

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

InitialSyncTransport

class torchrl.weight_update.InitialSyncTransport(*args, **kwargs)

Optional initial-connection capability for weight transports.

setup_connection_and_weights_on_receiver(*, worker_idx: int, weights: Any = None, model: Any = None, strategy: WeightStrategy | None = None) → Any

Synchronize weights on the receiver before collection starts.

setup_connection_and_weights_on_sender() → None

Synchronize weights on sender side before collection starts.

This is called once after workers are initialized to send the initial weights. This can be a no-op (weights are sent via send_weights).