Commit 9beb794e authored by AUTOMATIC's avatar AUTOMATIC
Browse files

clarify the option to disable NaN check.

parent 6f31d221

modules/devices.py

+2 −0
Original line number Diff line number Diff line
@@ -143,6 +143,8 @@ def test_for_nans(x, where): 
    else:

        message = "A tensor with all NaNs was produced."



    message += " Use --disable-nan-check commandline argument to disable this check."



    raise NansException(message)

CRA Git | Maintained and supported by SUSTech CRA and CCSE