Пробные тесты Tensorflow

Есть ли способ запустить тесты модулей Tensorflow вручную? Я хочу выполнить проверки работоспособности при изменении исходного кода TF.

Я вижу, что есть много файлов _test.py с классами, которые выполняют множество тестовых операций, и я не могу понять, как их запускать. Должен быть простой способ?

Ответы

Ответ 1

Самый простой способ запуска модульных тестов 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/...