人工智能:PyTorch深度学习框架介绍

icegoblin
发布于 2023-8-10 15:28
浏览
0收藏

今天给大家讲解一下PyTorch深度学习框架的一些基础知识,希望对大家理解PyTorch有一定的帮助!

人工智能:PyTorch深度学习框架介绍-鸿蒙开发者社区

1、PyTorch

PyTorch是一个基于Torch的Python机器学习框架。它是由Facebook的人工智能研究小组在2016年开发的,解决了Torch因为使用Lua编程语言普及度不高的问题,故采用了集成非常广泛的Python编程语言来实现。

2、PyTorch常用的工具包

● torch:类似于Numpy的通用数组库,可以在将张量类型转换为(torch.cuda.TensorFloat)并支持在GPU上进行计算。

● torch.autograd:主要用于构建计算图形并自动获取渐变的包

● torch.nn:具有共同层和成本函数的神经网络库

● torch.optim:具有通用优化算法(如SGD,Adam等)的优化包

● torch.utils:数据载入器。具有训练器和其他便利功能

● torch.legacy(.nn/.optim) :处于向后兼容性考虑,从 Torch 移植来的 legacy 代码

● torch.multiprocessing:python 多进程并发,实现进程之间 torch Tensors 的内存共享

3、PyTorch特点

● 动态神经结构:PyTorch通过一种反向自动求导的技术,可以做到零延迟地任意改变神经网络的行为,避免因为构建神经网络完成后,如果需要调整神经网络结构,只能从头开始的麻烦,采用PyTorch大大节省了人力和时间成本。

● Debug调试方便:PyTorch 的设计思路是线性、直观且易于使用的,当你执行一行代码时,一步步去执行,不涉及异步调试的繁琐,所以当代码出现Bug 的时候,可以通过很方便定位代码的位置,避免因为bug错误的指向或者异步和不透明的引擎造成查询问题的麻烦。

● 代码简洁易懂:PyTorch的代码相对于Tensorflow更加简洁容易读懂,并且PyTorch本身源代码阅读起来也要友好很多这样更容易深入理解PyTorch。

● 社区活跃度高:PyTorch 拥有一个非常活跃的社区和论坛(discuss.pytorch.org)。它的文档(pytorch.org)梳理的非常清晰,初学者入门很快;并且它与 PyTorch 版本保持同步,并提供一组完整教程。PyTorch 使用起来非常简单,所以学习成本也就比较低。

4、PyTorch不足之处

● 可视化监控能力弱:缺少直接可用监控和可视化接口:

● 不如TensorFlow应用广泛:PyTorch 不是端到端的机器学习开发工具;实际应用程序的开发需要将 PyTorch 代码转换为另一个框架,例如 Caffe2,转换后将应用程序部署到服务器、工作站和移动设备。


文章转载自公众号:小明互联网技术分享社区

分类
标签
已于2023-8-10 15:28:06修改
收藏
回复
举报
回复
    相关推荐