提交 6c4738d6 authored 作者: Frederic's avatar Frederic 提交者: Arnaud Bergeron

Make NanGuardMode not raise an error with theano.sandbox.rng_mrg.GPU_mrg_uniform

上级 fdbe417b
...@@ -179,6 +179,9 @@ class NanGuardMode(Mode): ...@@ -179,6 +179,9 @@ class NanGuardMode(Mode):
if nan_is_error: if nan_is_error:
err = False err = False
if cuda.cuda_available and isinstance(var, cuda.CudaNdarray): if cuda.cuda_available and isinstance(var, cuda.CudaNdarray):
if not isinstance(nd.op,
# It store ints in float container
theano.sandbox.rng_mrg.GPU_mrg_uniform):
err = np.isnan(self.gpumin(var.reshape(var.size))) err = np.isnan(self.gpumin(var.reshape(var.size)))
else: else:
err = contains_nan(var) err = contains_nan(var)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论