TransportBackend

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

Abstract interface for different communication mechanisms.

check_connection() → bool

Check if the connection is still alive.

receive_weights(timeout: float = 1.0) → tuple[str, Any] | None

Receive weights from the sender. Returns (model_id, weights) or None if timeout.

send_weights(model_id: str, weights: Any) → None

Send weights to the receiver.