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