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

Make the Gamma[Inplace]Tester don't crash with some seed.

上级 c9b5e6ba
......@@ -1397,12 +1397,8 @@ _good_broadcast_unary_gammaln = dict(
normal=(rand_ranged(-1 + 1e-2, 10, (2, 3)),),
empty=(numpy.asarray([]),),)
_grad_broadcast_unary_gammaln = dict(
normal=(rand_ranged(1e-8, 10, (2, 3)),),)
if theano.config.floatX == 'float32':
gamma_eps = 3e-4
else:
gamma_eps = 2e-10
# smaler range as our grad method don't estimate it good enough.
normal=(rand_ranged(1e-8, 8, (2, 3)),),)
GammaTester = makeBroadcastTester(
op=tensor.gamma,
......@@ -1410,7 +1406,6 @@ GammaTester = makeBroadcastTester(
good=_good_broadcast_unary_gammaln,
grad=_grad_broadcast_unary_gammaln,
mode=mode_no_scipy,
eps=gamma_eps,
skip=skip_scipy)
GammaInplaceTester = makeBroadcastTester(
op=inplace.gamma_inplace,
......@@ -1418,7 +1413,6 @@ GammaInplaceTester = makeBroadcastTester(
good=_good_broadcast_unary_gammaln,
grad=_grad_broadcast_unary_gammaln,
mode=mode_no_scipy,
eps=gamma_eps,
inplace=True,
skip=skip_scipy)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论