def parameters(self):
if self._parameters is None:
policy, baseline = self.policy, self.baseline
params = policy.trainable_variables
if baseline:
params = params + baseline.trainable_variables
params = U.unique(params)
self._parameters = params
return self._parameters
def parameters(self):
if self._parameters is None:
params = self.policy.trainable_variables + self.critic.trainable_variables
params = U.unique(params)
self._parameters = params
return self._parameters