def initialize(self, model): self.model = model self.variables = models.trainable_variables(self.model.critic) self.optimizer = self.optimizer(self.variables)
def initialize(self, model): self.model = model self.variables = models.trainable_variables(self.model.actor)
def initialize(self, model): self.model = model variables_1 = models.trainable_variables(self.model.critic_1) variables_2 = models.trainable_variables(self.model.critic_2) self.variables = variables_1 + variables_2 self.optimizer = self.optimizer(self.variables)