nntrainer
nntrainer copied to clipboard
[Wait for #2568] [ Tensor ] add is_NaN check in Tensor @open sesame 05/10 14:17
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:
- Build test: [X]Passed [ ]Failed [ ]Skipped
- Run test: [X]Passed [ ]Failed [ ]Skipped
Signed-off-by: jijoong.moon [email protected]