深度学习笔记4-tensorflow和pytorch

深度学习笔记4-tensorflow和pytorch

TensorFlow由谷歌大脑开发,并且在谷歌公司中广泛地应用于研究和生产需求。PyTorch是Torch框架的表亲,Torch是基于lua开发的,在Facebook公司里被广泛使用。

(1) TensorFlow被许多研究人员和行业专业人士使用。该框架的文档很齐全,社区活跃,碰到问题基本上都可以在社区中得到解决。Pytorch近几年才被提出,使用度比Tensorflow小,社区要小点,但增长趋势很猛。

(2) TensorFlow需要在模型能够运行之前静态地定义图。与外部世界的所有通信都通过tf.Session对象和tf.Placeholder来执行,这两个张量在运行时会被外部数据替代。在PyTorch中,图的定义则更为动态化:你可以随时定义、随时更改、随时执行节点,并且没有特殊的会话接口或占位符。总体而言,该框架与Python语言集成地更为紧密,并且在大多数时候用起来感觉更加本地化。而在用TensorFlow开发的时候,你会觉得模型是在一堵墙的后面,仅能通过墙上的几个小洞与之通讯。

(3) Tensorflow包含Tensorboard,可视化做的非常好。Pytorch一般是用matplotlib和seaborn来进行可视化。

(4) TensorFlow 支持移动和嵌入式部署,而Pytorch没有这个能力。此外,TensorFlow Serving支持高性能的服务器端部署。

(5) PyTorch像一个框架,可用nn.Module创建复杂的深度学习架构。而纯TensorFlow看起来更像是一个库,而不是框架:所有的操作都在低层次进行,因此你不得不编写大量的样板代码(偏差和权重定义等等)。但现在TensorFlow高层次包装的整个生态环境开始出现,在选择适合任务的框架上有着很大的自由度。

——————————
PyTorch 更适用于研究、爱好者和小规模项目的快速原型开发。TensorFlow 更适合大规模部署,尤其是涉及跨平台和嵌入式部署时。

但就个人而言,我更推荐Pytorch。PyTorch更为清晰,更为python化,让开发者有更好的开发和调试体验。它的nn.Module让你能够以面向对象的方式来定义可重用的模块,这种方法非常灵活非常强大。然后,你可以使用nn.Sequential来组成各种模块。此外,你还可以以函数的形式使用所有的内置模块,这非常方便。总的来说,API的各个方面用起来都很顺手。

参考:
PyTorch vs TensorFlow,哪个更适合你
PyTorch和TensorFlow哪家强:九项对比读懂各自长项短板

猜你喜欢

转载自blog.csdn.net/zhennang1427/article/details/85396958