Skip to main content
 Web开发网 » 站长学院 » 浏览器插件

面对Tensorflow,为何我选择PyTorch?

2021年11月04日5670百度已收录

先来说说pytorch劣势。自从其被发布以来pytorch更多被用于学术界而不是工业界的实际生产,主要是因为它不够成熟、很多接口不太稳定,加上其全面性也不够,tensorflow目前仍然有很多pytorch不支持的功能,比如快速傅里叶变换,但这一点劣势会随着pytorch的发展而逐渐减小。除此此外,相比于tensorflow的容易各处部署的静态图(这一点远胜于很多框架),以python优先的深度学习框架 pytorch在部署到其他产品会很不方便。

优势先从上手时间开始说,虽然在2015年发布之后tensorflow多方受宠,但是和theano一样,tensorflow使用的是静态计算图,对于新手来说有过多需要新学习的概念,这导致了不管是入门还是搭建,使用tensorflow都比pytorch困难。而在2017年pytorch被团队开源的主要原因之一也是让建立深度学习模型更加简单,这让它发展十分迅猛。在数据加载上,pytorch加载数据的API简单高效,其面向对象的API源自于porch(也是keras的设计起源),比tensorflow难学的API友好很多,使用户可以将重点放在实现自己的想法,而不是被框架本身束缚住。

速度上,pytorch并没有为了灵活性而放弃速度,虽然运行速度和程序员水平密切相关,但pytorch在相同情况下常有可能胜于其他框架的速度。另外,如果追求自定义拓展,pytorch也会是首选,因为虽然二者的构建和绑定有一定的相似点,但是tensorflow在拓展时会需要很多样板代码,但pytorch只用编写接口和实现。

评论列表暂无评论
发表评论
微信