nntrainer icon indicating copy to clipboard operation
nntrainer copied to clipboard

[Wait for #2568] [ Tensor ] add is_NaN check in Tensor @open sesame 05/10 14:17

Open jijoongmoon opened this issue 2 months ago • 9 comments

In this PR

This PR adds the is_NaN function to check if the tensor has a NaN value. This is for the check of NaN during mixed precision training.

Self evaluation:

  1. Build test: [X]Passed [ ]Failed [ ]Skipped
  2. Run test: [X]Passed [ ]Failed [ ]Skipped

Signed-off-by: jijoong.moon [email protected]

jijoongmoon avatar May 08 '24 10:05 jijoongmoon