基于TensorFlow和PyTorch的深度学习框架对比分析

摘要：

深度学习框架是实现机器学习的关键工具,合适的深度学习框架可以达到事半功倍的效果.为助力研究者选择合适的框架,在回顾近十种常见框架的基础上,聚焦当前受众最广、热度最高的两种深度学习框架TensorFlow和PyTorch,从历程、现状、机制、训练模式、可视化、工业部署等角度对两者进行比对分析,并归类对应适用场景的建议,为框架选择提供思路参考.

关键词：深度学习、TensorFlow、PyTorch、适用场景

所属期刊栏目：4

分类号：TP181;TP391.41(自动化基础理论)

资助基金：广东交通职业技术学院大学生科技创新项目;广东交通职业技术学院一院一坊项目;2019年广东交通职业技术学院大学生创新创业训练计划项目

在线出版日期：2020-05-18（万方平台首次上网日期，不代表论文的发表时间）

页数：共4页

页码：80-82,87