get_flat_parameters

class maze.train.trainers.es.es_utils.get_flat_parameters(policy: maze.core.agent.torch_policy.TorchPolicy)

Get the parameters of all sub-policies as a single flat vector.

Parameters

policy – source policy

Returns

flattened parameters