def _post_load(self): if self._optimizer is not None: update_optimizer_parameters(self._optimizer, list(self.policy.parameters()))
def _update_optimizer_parameters(self, parameters): self._parameters = list(parameters) if self._optimizer is not None: update_optimizer_parameters(self._optimizer, self._parameters)