set_flat_parameters

class maze.train.trainers.es.es_utils.set_flat_parameters(policy: maze.core.agent.torch_model.TorchModel, flat_params: torch.Tensor)

Overwrite the parameters of all sub-policies by a single flat vector.

Parameters

• policy – target policy

• flat_params – concatenated vector