浅谈深度学习框架---Tensorflow、pytorch、caffe、keras

      神经网络的层数越深,模型训练加上反向传播的代码实现就越繁琐,神经网络的框架就很有必要。框架的作用就是让你简化代码,提升开发效率。

      Theano是早期的深度学习框架,但是这个框架已经停止开发。caffe与keras是高层的深度学习框架可以快速地验证模型,Tensorflow与pytorch是底层的深度学习框架可以实现对神经网络底层细节的修改。Tensorflow由谷歌团队开发,keras可以看作是Tensorflow封装的高级API。caffe与pytorch是由facebook开发。Tensorflow的社区非常友好庞大,对于初学者来说遇到问题很容易找到解决方案。keras是比较简单易学的,因为它的高封装性。torch是动态图计算,其余三种是静态图计算,需要先定义再使用。Tensorflow的部署比较方便,可以使用C++或者python或者前端的JS。

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐