ray.rllib.algorithms.algorithm.Algorithm.set_weights#

Algorithm.set_weights(weights: Dict[str, dict])[source]#

Set RLModule/Policy weights by Module/Policy ID.

Parameters:

weights – Dict mapping ModuleID/PolicyID to weights.