提交 8ff173ff authored 作者: hantek's avatar hantek

PEP8 respect

上级 1376305e
...@@ -115,6 +115,7 @@ class NanGuardMode(Mode): ...@@ -115,6 +115,7 @@ class NanGuardMode(Mode):
[self.guard_input], T.max(T.abs_(self.guard_input)), [self.guard_input], T.max(T.abs_(self.guard_input)),
mode='FAST_RUN' mode='FAST_RUN'
) )
def do_check_on(var, nd, f, is_input): def do_check_on(var, nd, f, is_input):
""" """
Checks `var` for NaNs / Infs. If detected, raises an exception Checks `var` for NaNs / Infs. If detected, raises an exception
...@@ -146,7 +147,7 @@ class NanGuardMode(Mode): ...@@ -146,7 +147,7 @@ class NanGuardMode(Mode):
if inf_is_error: if inf_is_error:
err = False err = False
if cuda.cuda_available and isinstance(var, cuda.CudaNdarray): if cuda.cuda_available and isinstance(var, cuda.CudaNdarray):
err = (np.isinf(self.gpumin(var.reshape(var.size))) or \ err = (np.isinf(self.gpumin(var.reshape(var.size))) or
np.isinf(self.gpumax(var.reshape(var.size)))) np.isinf(self.gpumax(var.reshape(var.size))))
else: else:
err = contains_inf(var) err = contains_inf(var)
...@@ -191,7 +192,8 @@ class NanGuardMode(Mode): ...@@ -191,7 +192,8 @@ class NanGuardMode(Mode):
The thunk to execute for this Apply node The thunk to execute for this Apply node
""" """
inputs = fn.inputs inputs = fn.inputs
# TODO: figure out why individual inputs are themselves lists sometimes # TODO: figure out why individual inputs are themselves lists
# sometimes
for x in flatten(inputs): for x in flatten(inputs):
do_check_on(x, node, fn, True) do_check_on(x, node, fn, True)
fn() fn()
...@@ -199,5 +201,7 @@ class NanGuardMode(Mode): ...@@ -199,5 +201,7 @@ class NanGuardMode(Mode):
for j, x in enumerate(flatten(outputs)): for j, x in enumerate(flatten(outputs)):
do_check_on(x, node, fn, False) do_check_on(x, node, fn, False)
wrap_linker = theano.gof.WrapLinkerMany([theano.gof.OpWiseCLinker()], [nan_check]) wrap_linker = theano.gof.WrapLinkerMany([theano.gof.OpWiseCLinker()],
super(NanGuardMode, self).__init__(wrap_linker, optimizer=theano.config.optimizer) [nan_check])
super(NanGuardMode, self).__init__(wrap_linker,
optimizer=theano.config.optimizer)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论