pytorch_pfn_extras.training.extensions.fail_on_non_number#
Classes
|
|
|
An extension to raise RuntimeError if parameters and its gradients contain NaN or Inf. |
Site Navigation
Section Navigation
Classes
|
|
|
An extension to raise RuntimeError if parameters and its gradients contain NaN or Inf. |