运行Tensorflow单元测试
tensorflow
7
0

有什么方法可以手动运行Tensorflow单元测试吗?我想在修改TF源代码时执行健全性检查。

我看到有许多_test.py文件,它们具有执行许多测试操作的类,但我不知道如何运行它们。应该有一个简单的方法吗?

参考资料:
Stack Overflow
收藏
评论
共 2 个回答
高赞 时间 活跃

除了上述答案之外,您还可以运行如图所示的单个测试,而不是运行完整的程序包,这样可以节省大量时间。

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
收藏
评论

假设您已从Git下载源代码 ,那么运行TensorFlow单元测试的最简单方法是使用Bazel:

# 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/...
收藏
评论
新手导航
  • 社区规范
  • 提出问题
  • 进行投票
  • 个人资料
  • 优化问题
  • 回答问题

关于我们

常见问题

内容许可

联系我们

@2020 AskGo
京ICP备20001863号