在 TensorFlow 中调试 Python 测试

问题描述:

我们想在 TensorFlow 中调试 Python 测试,例如 sparse_split_op_teststring_to_hash_bucket_op_test

We want to debug Python tests in TensorFlow such as sparse_split_op_test and string_to_hash_bucket_op_test

我们可以使用 gdb 调试其他 C++ 测试,但是我们找不到调试 python 测试的方法.

The other c++ tests we could debug using gdb, however we cannot find a way to debug python tests.

有没有一种方法可以调试通过 Bazel 测试命令运行的特定 python 测试用例(例如,bazel test//tensorflow/python/kernel_tests:sparse_split_op_test)

Is there a way in which we can debug specific python test case run via Bazel test command (for example, bazel test //tensorflow/python/kernel_tests:sparse_split_op_test)

我会先构建测试:

bazel build //tensorflow/python/kernel_tests:sparse_split_op_test

然后在生成的 Python 二进制文件上使用 pdb:

Then use pdb on the resulting Python binary:

pdb bazel-bin/tensorflow/python/kernel_tests/sparse_split_op_test

这似乎对我逐步完成测试的前几行有用.

That seems to work for me stepping through the first few lines of the test.