Enable all GPU CTC tests

上级 9ac0c447
...@@ -26,11 +26,11 @@ class TestCTC(unittest.TestCase): ...@@ -26,11 +26,11 @@ class TestCTC(unittest.TestCase):
inputs = [t_activations, t_labels, t_activation_times] inputs = [t_activations, t_labels, t_activation_times]
# Execute several tests for each test case # Execute several tests for each test case
#self.check_expected_values(t_activations, t_labels, t_activation_times, expected_costs, expected_grads) self.check_expected_values(t_activations, t_labels, t_activation_times, expected_costs, expected_grads)
#self.compare_gpu_and_cpu_values(*inputs) self.compare_gpu_and_cpu_values(*inputs)
#self.check_grads_disabled(*inputs) self.check_grads_disabled(*inputs)
self.run_gpu_optimization_with_grad(*inputs) self.run_gpu_optimization_with_grad(*inputs)
#self.run_gpu_optimization_no_grad(*inputs) self.run_gpu_optimization_no_grad(*inputs)
def setup_cpu_op(self, activations, labels, input_length, compute_grad=True, mode=mode_without_gpu): def setup_cpu_op(self, activations, labels, input_length, compute_grad=True, mode=mode_without_gpu):
cpu_ctc_cost = ctc(activations, labels, input_length) cpu_ctc_cost = ctc(activations, labels, input_length)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论