提交 a1bb429e authored 作者: Olivier Delalleau's avatar Olivier Delalleau

Removed warning when running tests

上级 56f17d7e
...@@ -57,6 +57,9 @@ class T_sigmoid_opts(unittest.TestCase): ...@@ -57,6 +57,9 @@ class T_sigmoid_opts(unittest.TestCase):
x = T.vector() x = T.vector()
data = numpy.random.rand(54).astype(config.floatX) data = numpy.random.rand(54).astype(config.floatX)
backup = config.warn.identify_1pexp_bug
config.warn.identify_1pexp_bug = False
try:
# tests exp_over_1_plus_exp # tests exp_over_1_plus_exp
f = theano.function([x], T.exp(x) / (1 + T.exp(x)), mode=m) f = theano.function([x], T.exp(x) / (1 + T.exp(x)), mode=m)
assert [node.op for node in f.maker.env.toposort()] == [sigmoid] assert [node.op for node in f.maker.env.toposort()] == [sigmoid]
...@@ -138,6 +141,10 @@ class T_sigmoid_opts(unittest.TestCase): ...@@ -138,6 +141,10 @@ class T_sigmoid_opts(unittest.TestCase):
T.mul, theano.tensor.inplace.neg_inplace] T.mul, theano.tensor.inplace.neg_inplace]
f(data) f(data)
finally:
# Restore config option.
config.warn.identify_1pexp_bug = backup
def test_1msigmoid(self): def test_1msigmoid(self):
if not register_local_1msigmoid: if not register_local_1msigmoid:
return return
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论