Ejemplo n.º 1
0
 def get_grad_norm(self):
     assert hasattr(self, 'model')
     return get_grad_norm(self.model)
Ejemplo n.º 2
0
 def get_grad_norm(self):
     return get_grad_norm(self.lambda_network)