最近想入门深度学习,本来打算直接学PyTorch,但看到很多公司招聘要求TensorFlow,又犹豫了。自己跟着教程跑过几个demo,PyTorch的调试确实直观,但一到部署就懵了,ONNX转换各种报错。TensorFlow 2.x的Keras接口也挺简洁,但版本兼容问题搞到头大。想问问大家实际工作中到底用哪个多?如果目标是明年找算法岗,现在应该主攻哪个框架?还是说两个都得会一点?求过来人指点。
楼主
2天前
PyTorch和TensorFlow学哪个?感觉教程看越多越迷茫了
请 登录 后发表回复
全部回复
共 1 条
2楼
1天前
说个实在的,你要是明年找算法岗,现在纠结这个纯粹是浪费时间。我面过不少人,简历上写精通TF结果连动态图静态图都分不清的太多了。PyTorch现在学术界基本是统治地位,你去看顶会论文的代码,十篇里有八篇是PyTorch,面试官问你最近复现过什么模型,你用TF讲起来自己都别扭。TensorFlow在企业里确实存量多,但很多是历史项目,新起的团队尤其是做大模型相关的,PyTorch占绝对优势。部署那块ONNX报错太正常了,跟框架关系不大,是你对模型导出时的动态轴、算子支持没吃透,换个框架照样踩坑。我的建议是主攻PyTorch,把训练和调试链路跑熟,然后花两周了解一下TF的Keras和SavedModel格式就行,面试问到能说清楚区别就够了。两个都学一点这种话听着稳妥,实际上容易两边都不深,面试官一追问就露馅。