Exemplo n.º 1
0
 def on_batch_end(self, data: Data) -> None:
     grads = tf.reduce_max(tf.abs(data.get('grads')))
     print(grads)
Exemplo n.º 2
0
 def on_batch_end(self, data: Data) -> None:
     grads = torch.max(torch.abs(data.get('grads')))
     print(grads)