提交 f78fd294 authored 作者: Frederic's avatar Frederic

[CRASH] fix gh-3447

上级 da3c8070
...@@ -263,14 +263,14 @@ class NanGuardMode(Mode): ...@@ -263,14 +263,14 @@ class NanGuardMode(Mode):
error = True error = True
if big_is_error: if big_is_error:
err = False err = False
if var.size == 0: if isinstance(var, theano.gof.type.CDataType._cdata_type):
err = False err = False
elif cuda.cuda_available and isinstance(var, cuda.CudaNdarray): elif cuda.cuda_available and isinstance(var, cuda.CudaNdarray):
err = (f_gpuabsmax(var.reshape(var.size)) > 1e10) err = (f_gpuabsmax(var.reshape(var.size)) > 1e10)
elif isinstance(var, theano.gof.type.CDataType._cdata_type):
err = False
elif isinstance(var, np.random.mtrand.RandomState): elif isinstance(var, np.random.mtrand.RandomState):
err = False err = False
elif var.size == 0:
err = False
else: else:
err = (np.abs(var).max() > 1e10) err = (np.abs(var).max() > 1e10)
if err: if err:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论