unilab.ipc.weight_sync.SharedWeightSync

class unilab.ipc.weight_sync.SharedWeightSync[source]

Bases: object

Synchronize actor weights between learner and collector.

Parameters:

Methods

__init__(param_shapes, *[, create, ...])

cleanup()

close()

from_state_dict(state_dict, **kwargs)

read_weights_into(state_dict)

write_weights(state_dict)

Attributes

name

version

__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