def named_mixing_parameters(self, **kwargs):
"""Returns an iterator over mixing module parameters, yielding both the
name of the parameter as well as the parameter itself
"""
return ptu.named_parameters(self._mixing_modules,
self._mixing_parameters,
**kwargs)
def named_policies_parameters(self, idx=None, **kwargs):
"""Returns an iterator over policies module parameters, yielding both the
name of the parameter as well as the parameter itself
"""
if idx is None:
idx_list = list(range(self._n_subpolicies))
elif isinstance(idx, list) or isinstance(idx, tuple):
idx_list = idx
else:
idx_list = [idx]
return chain(*[ptu.named_parameters(self._policies_modules[idx],
self._policies_parameters[idx],
**kwargs)
for idx in idx_list])
def named_final_policy_parameters(self, **kwargs):
"""Returns an iterator over final policy module parameters, yielding
both the name of the parameter as well as the parameter itself
"""
return ptu.named_parameters(self._final_policy_modules,
self._final_policy_parameters, **kwargs)