WeightReceiver

class torchrl.weight_update.WeightReceiver(scheme: WeightSyncScheme)

Receives weights for ONE model in ONE worker.

This class handles receiving weights via transport and applying them to a registered model in the worker process.

apply_weights(weights: Any) → None

Apply received weights to registered model.

Parameters:

weights – The weights to apply.

register_model(model_ref: Any) → None

Register the model to apply weights to.

Parameters:

model_ref – Either a direct object reference or a string path like ‘policy’ or ‘env.value_net’.

register_worker_transport(pipe: Any) → None

Register this worker’s communication pipe.

Parameters:

pipe – The pipe connection for this worker.

set_context(context: Any) → None

Set the context object (inner_collector) for resolving references.

Parameters:

context – The inner collector instance in the worker process.