Remove unused function from CTC GPU wrapper tests

上级 f5e36392
...@@ -27,10 +27,8 @@ class TestCTC(unittest.TestCase): ...@@ -27,10 +27,8 @@ class TestCTC(unittest.TestCase):
t_grad = T.grad(T.mean(t_cost), t_activations) t_grad = T.grad(T.mean(t_cost), t_activations)
# Compile symbolic functions # Compile symbolic functions
train = theano.function([], [t_cost, t_grad]) train = theano.function([], [t_cost, t_grad])
test = theano.function([], [t_cost])
cost, grad = train() cost, grad = train()
cost, = test()
cpu_cost = np.empty(shape=cost.shape, dtype=np.float32) cpu_cost = np.empty(shape=cost.shape, dtype=np.float32)
# Transfer costs from GPU memory to host # Transfer costs from GPU memory to host
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论