提交 60bb99bf authored 作者: Frederic Bastien's avatar Frederic Bastien

fix test_autocast in floatX=float32

上级 0e0f4802
...@@ -2447,7 +2447,7 @@ def test_autocast(): ...@@ -2447,7 +2447,7 @@ def test_autocast():
ac = autocast_float_as('float32', 'float64') ac = autocast_float_as('float32', 'float64')
ac.__enter__() ac.__enter__()
assert (dvector()+ 1.1).dtype == 'float64' assert (dvector()+ 1.1).dtype == 'float64'
assert (fvector()+ 1.1).dtype == 'float64' assert (fvector()+ 1.1).dtype == theano.config.floatX
assert (fvector()+ 1.0).dtype == 'float32' assert (fvector()+ 1.0).dtype == 'float32'
try: #ghetto 2.4 version of with try: #ghetto 2.4 version of with
ac2 = autocast_float_as('float64') ac2 = autocast_float_as('float64')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论