MultiProcessWeightSyncScheme

class torchrl.weight_update.MultiProcessWeightSyncScheme(strategy: Literal['state_dict', 'tensordict'] = 'state_dict')

Weight synchronization for multiprocess operations using pipes.

This scheme creates transports that communicate via multiprocessing pipes.

create_receiver() → WeightReceiver

Create a receiver for this scheme.

Returns:

WeightReceiver instance configured for this scheme.

create_sender() → WeightSender

Create a sender for this scheme.

Returns:

WeightSender instance configured for this scheme.

create_transport(pipe: Any) → TransportBackend

Create an MPTransport using the provided pipe.