unilab.ipc.weight_sync
Shared weight synchronization for actor networks.
Classes
-
class unilab.ipc.weight_sync.SharedWeightSync[source]
Bases: object
Synchronize actor weights between learner and collector.
- Parameters:
-
-
__init__(param_shapes, *, create=True, shm_name=None, lock=None)[source]
- Parameters:
-
-
property name: str
-
property version: int
-
classmethod from_state_dict(state_dict, **kwargs)[source]
-
write_weights(state_dict)[source]
- Return type:
None
-
read_weights_into(state_dict)[source]
- Return type:
int
-
cleanup()[source]
- Return type:
None
-
close()[source]
- Return type:
None