Esiste un modo per eseguire i test dell'unità Tensorflow manualmente? Voglio eseguire controlli di integrità durante la modifica del codice sorgente TF.Esegui test unità Tensorflow
Vedo che ci sono molti file _test.py con classi che eseguono molte operazioni di test e non riesco a capire come eseguirli. Ci dovrebbe essere un modo semplice?
Il modo più semplice per eseguire i test di unità tensorflow sta usando Bazel, a patto di avere 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/...
E come si dovrebbe eseguire questi test con supporto GPU? Solo eseguendoli direttamente sembra che utilizzi solo la CPU: 'I tensorflow/core/common_runtime/local_device.cc: 40] Thread di parallelismo intra op del dispositivo locale: 16 I tensorflow/core/common_runtime/direct_session.cc: 58] Interoperazione di sessione diretta thread di parallelismo: 16 ' – panmari
Ho aggiornato la risposta per mostrare come creare ed eseguire con il supporto della GPU. (Si noti che quei messaggi di registro saranno prodotti nelle modalità CPU * e * GPU.) – mrry
Certo, volevo solo sottolineare che solo la CPU è elencata in questo momento. Grazie per il tuo aggiornamento! – panmari