Skip to content

P
pytensor
提交 fa55c51b authored 作者: Razvan Pascanu's avatar Razvan Pascanu
浏览文件
操作

decreased the sizes of the matrices to make debug mode run faster on those

tests.
上级 953e2f77
隐藏空白字符变更
内嵌 并排
正在显示 包含 6 行增加 和 6 行删除
theano/tests/test_tutorial.py
浏览文件 @ fa55c51b
...@@ -763,8 +763,8 @@ class T_using_gpu(unittest.TestCase): ...@@ -763,8 +763,8 @@ class T_using_gpu(unittest.TestCase):
import numpy import numpy
import time import time
vlen = 10 * 30 * 768 # 10 x #cores x # threads per core vlen = 10 * 30 * 70 # 10 x #cores x # threads per core
iters = 1000 iters = 10
rng = numpy.random.RandomState(22) rng = numpy.random.RandomState(22)
x = shared(numpy.asarray(rng.rand(vlen), config.floatX)) x = shared(numpy.asarray(rng.rand(vlen), config.floatX))
...@@ -791,8 +791,8 @@ class T_using_gpu(unittest.TestCase): ...@@ -791,8 +791,8 @@ class T_using_gpu(unittest.TestCase):
import numpy import numpy
import time import time
vlen = 10 * 30 * 768 # 10 x #cores x # threads per core vlen = 10 * 30 * 70 # 10 x #cores x # threads per core
iters = 1000 iters = 10
rng = numpy.random.RandomState(22) rng = numpy.random.RandomState(22)
x = shared(numpy.asarray(rng.rand(vlen), config.floatX)) x = shared(numpy.asarray(rng.rand(vlen), config.floatX))
...@@ -820,8 +820,8 @@ class T_using_gpu(unittest.TestCase): ...@@ -820,8 +820,8 @@ class T_using_gpu(unittest.TestCase):
import numpy import numpy
import time import time
vlen = 10 * 30 * 768 # 10 x #cores x # threads per core vlen = 10 * 30 * 70 # 10 x #cores x # threads per core
iters = 1000 iters = 10
rng = numpy.random.RandomState(22) rng = numpy.random.RandomState(22)
x = shared(numpy.asarray(rng.rand(vlen), config.floatX)) x = shared(numpy.asarray(rng.rand(vlen), config.floatX))
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论