我對Keras與其他TF包裝(如Tensorpack)的培訓性能有疑問。TF包裝:Keras與Tensorpack的性能
在a Q&A session,Keras的作者說,包裝內附送的性能沒有影響:
Keras能夠構建一個簡單,快捷的方式,並在TensorFlow火車模型,在性能沒有影響,因爲該機型仍然由相同的TensorFlow引擎運行。
在另一方面雖然,Tensorpack github聲稱它比Keras快2倍,即使一個小小的CNN:
Tensorpack教練幾乎總是比基於feed_dict包裝速度更快。即使在一個微小的CNN例子中,訓練運行速度比同等Keras代碼快兩倍。
所以現在我想知道以下幾點:誰是對的? 對於我來說,訓練中兩個因素是非常重要的。
你自己試過嗎? –
是的,我一直在使用Keras(雖然從未使用過Tensorpack)。我只關心性能,因爲我有4D數據集,數百GB,一個'圖像'是11x13x18x50。目前,我在單個節點中使用Keras TF後端和4個GTX1080(現在多節點不能很好地擴展)。 – 0vbb