def update_targets(self):
     polyak_update(self.actor.parameters(),
                   self.actor_target.parameters(),
                   tau=0.005)
     polyak_update(self.critic1.parameters(),
                   self.critic_target1.parameters(),
                   tau=0.005)
예제 #2
0
 def update_targets(self):
     polyak_update(self.critic1.parameters(),
                   self.critic_target1.parameters(),
                   tau=self.tau)
     polyak_update(self.critic2.parameters(),
                   self.critic_target2.parameters(),
                   tau=self.tau)