web-dev-qa-db-ja.com

Tensorflow単体テストを実行する

Tensorflowユニットテストを手動で実行する方法はありますか？ TFのソースコードを変更しながら、健全性チェックを実行したい。

多くのテスト操作を実行するクラスを含む多くの_test.pyファイルがあり、それらを実行する方法を理解できません。簡単な方法があるはずですか？

unit-testingtensorflowtestingbazel
24
Maksym Diachenko

TensorFlowユニットテストを実行する最も簡単な方法は、Bazelを使用することです 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/...
35
mrry

上記の回答に加えて、完全なパッケージの代わりに、示されているように個々のテストを実行できます。これにより、時間を大幅に節約できます。

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
5
Lucas Hendren