Esempi in Python per try_store_grad_norm

Esempio n. 1

 def update(self):
     net_util.try_store_grad_norm(self)
     self.body.explore_var = self.explore_var_scheduler.update(
         self, self.body.env.clock)
     if self.entropy_coef_spec is not None:
         self.body.entropy_coef = self.entropy_coef_scheduler.update(
             self, self.body.env.clock)
     return self.body.explore_var

Esempio n. 2

File: sarsa.py Progetto: wilson1yan/SLM-Lab

 def update(self):
     '''Update the agent after training'''
     net_util.try_store_grad_norm(self)
     self.body.explore_var = self.explore_var_scheduler.update(
         self, self.body.env.clock)
     return self.body.explore_var