2015-12-10 5 views
16

क्या Tensorflow इकाई परीक्षण मैन्युअल रूप से चलाने का कोई तरीका है? मैं टीएफ स्रोत कोड को संशोधित करते समय सैनिटी चेक करना चाहता हूं।रन Tensorflow इकाई परीक्षण

मुझे लगता है कि कक्षाओं के साथ कई _test.py फाइलें हैं जो कई परीक्षण संचालन करती हैं और मैं यह नहीं समझ सकता कि उन्हें कैसे चलाया जाए। एक आसान तरीका होना चाहिए?

उत्तर

22

TensorFlow इकाई परीक्षण चलाने के लिए सबसे आसान तरीका है Bazel उपयोग कर रहा है, यह मानते हुए आप 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/... 
+0

और कैसे आप GPU समर्थन के साथ इन परीक्षणों को चलाने के हैं? बस उन्हें सीधे निष्पादित करने के लिए केवल सीपीयू का उपयोग लगता है: 'I tensorflow/core/common_runtime/local_device.cc: 40] स्थानीय डिवाइस इंट्रा ऑप समांतरता धागे: 16 मैं tensorflow/core/common_runtime/direct_session.cc: 58] डायरेक्ट सत्र इंटर सेशन समांतरता धागे: 16 ' – panmari

+0

मैंने GPU समर्थन के साथ निर्माण और चलाने के तरीके को दिखाने के लिए उत्तर अपडेट किया। (ध्यान दें कि उन लॉग संदेशों को सीपीयू * और * जीपीयू मोड में उत्पादित किया जाएगा।) – mrry

+0

निश्चित रूप से, मैं बस जोर देना चाहता था कि केवल सीपीयू ही सूचीबद्ध है। आपके अद्यतन के लिए धन्यवाद! – panmari

संबंधित मुद्दे