提交 9501ac8f authored 作者: Rebecca N. Palmer's avatar Rebecca N. Palmer

flake8 code style fixes

上级 0da749df
......@@ -261,7 +261,7 @@ class Mode(object):
def __init__(self, linker=None, optimizer='default'):
if linker is None:
linker = config.linker
if type(optimizer)==str and optimizer == 'default':
if type(optimizer) == str and optimizer == 'default':
optimizer = config.optimizer
Mode.__setstate__(self, (linker, optimizer))
......
......@@ -642,7 +642,7 @@ class AbstractBatchNormTrainGrad(Op):
# some inputs should be disconnected
results = [g_wrt_x, g_wrt_dy, g_wrt_scale, g_wrt_x_mean, g_wrt_x_invstd,
theano.gradient.DisconnectedType()()]
return [theano.gradient.DisconnectedType()() if (type(r)==int and r==0) else r
return [theano.gradient.DisconnectedType()() if (type(r) == int and r == 0) else r
for r in results]
def connection_pattern(self, node):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论