ray.rllib.algorithms.algorithm.Algorithm.get_weights#

Algorithm.get_weights(policies: List[str] | None = None) dict[source]#

Return a dict mapping Module/Policy IDs to weights.

Parameters:

policies – Optional list of policies to return weights for, or None for all policies.