我对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