运行 Tensorflow 单元测试
Run Tensorflow unit tests
有什么方法可以 运行 手动进行 Tensorflow 单元测试吗?我想在修改 TF 源代码时执行健全性检查。
我看到有许多 _test.py 文件 类 执行许多测试操作,但我不知道如何 运行 它们。应该有简单的方法吧?
运行TensorFlow 单元测试的最简单方法是使用 Bazel,假设您有 downloaded the source from Git:
# All tests (for C++ changes).
$ bazel test //tensorflow/...
# All Python tests (for Python front-end changes).
$ bazel test //tensorflow/python/...
# All tests (with GPU support).
$ bazel test -c opt --config=cuda //tensorflow/...
$ bazel test -c opt --config=cuda //tensorflow/python/...
除了上述答案之外,您还可以运行如图所示的单独测试而不是完整的包,这可以节省大量时间。
bazel run //tensorflow/python/kernel_tests:string_split_op_test
bazel run //tensorflow/python:special_math_ops_test
或者您可以转到单独的目录并运行那里的所有测试
cd python/kernel_tests
bazel run :one_hot_op_test
有什么方法可以 运行 手动进行 Tensorflow 单元测试吗?我想在修改 TF 源代码时执行健全性检查。
我看到有许多 _test.py 文件 类 执行许多测试操作,但我不知道如何 运行 它们。应该有简单的方法吧?
运行TensorFlow 单元测试的最简单方法是使用 Bazel,假设您有 downloaded the source from Git:
# All tests (for C++ changes).
$ bazel test //tensorflow/...
# All Python tests (for Python front-end changes).
$ bazel test //tensorflow/python/...
# All tests (with GPU support).
$ bazel test -c opt --config=cuda //tensorflow/...
$ bazel test -c opt --config=cuda //tensorflow/python/...
除了上述答案之外,您还可以运行如图所示的单独测试而不是完整的包,这可以节省大量时间。
bazel run //tensorflow/python/kernel_tests:string_split_op_test
bazel run //tensorflow/python:special_math_ops_test
或者您可以转到单独的目录并运行那里的所有测试
cd python/kernel_tests
bazel run :one_hot_op_test