Запустите модульные тесты 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/...
Источник

В дополнение к приведенному выше ответу вы можете запустить отдельные тесты, как показано, вместо полных пакетов, что может сэкономить значительное количество времени.

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
Источник
Другие вопросы по тегам