热天中文网

第623章 这框架简直是神器(第2页)

天才一秒记住【热天中文网】地址:https://www.rtzw.net

郭长征现在完全没有了困意。

楚一航滑动滑鼠,他们来到了另一个模块区。

“郭老师,你看这个,这里封装了好多高级模块。”

楚一航点开了一个名为卷积的目录。

里面列著各种奇怪的函数名。

两人仔细看著文档的解释。

里面有一个叫卷积神经网络的封装层。

两人平时研究的都是些传统的机器学习算法,对这种结构並不算特別清楚。

直到看到文档里清清楚楚地写著:该模块通过卷积核提取局部特徵,极为適用於图像识別与计算机视觉领域。

“图像识別?”

郭长征敏锐地抓住了这个词。

“我们下午吃饭的时候,討论的李飞飞教授那个imagenet数据集,不就是图像分类吗?”

楚一航一拍桌子:“绝了!

连这种专门针对图像的模块都提前封装好了。”

“这简直就是给咱们量身定製的武器库啊。”

两人对著这些功能模块,越討论越兴奋。

“一航,別光看著眼馋了,纸上得来终觉浅。”

郭长征提议。

“咱们现在就动手,按这个文档,写一个简单的几层网络试试水。”

“好嘞。”

楚一航立刻切换到代码编辑器。

按照平时用cuda编写底层算力代码的习惯,楚一航习惯性地准备先写一堆內存分配的语句。

在以往,用英伟达的cuda写代码,繁琐。

程式设计师必须手动在主机的內存里开闢空间,再在显卡的显存里开闢空间。

然后要把数据从主机复製到显卡,接著手动设置线程块和网格的大小。

稍微设置不合理,显卡就直接罢工不干。

算完之后,还得再把数据从显卡慢吞吞地拷回主机。

但当楚一航按照tensorflow的文档指引敲击键盘时,他发现那些全都不需要了。

他只需要定义好数据格式,声明好网络层数,框架的底层执行引擎会自动接管所有的硬体资源调度。

不到二十行代码。

简洁,优雅。

“这就写完了?”

楚一航自己都不敢相信。

“这几十行代码,放以前用c语言加cuda手搓,至少得写大半天,还得调试一整天。”

郭长征凑在屏幕前,反反覆覆检查了三遍代码逻辑。

本章未完,请点击下一章继续阅读!若浏览器显示没有新章节了,请尝试点击右上角↗️或右下角↘️的菜单,退出阅读模式即可,谢谢!

如遇章节错误,请点击报错(无需登陆)

新书推荐

诸天万界乱入者世子妃靠美食成了团宠娇娇女息桐重生之花都医仙峡谷没一个能打的吗营业悖论[娱乐圈]温柔沦陷我的修炼能加速重生成偏执霍少的小仙女大宇之上奋斗在瓦罗兰长姐穿越啦穿成清太孙的咸鱼日常这个小青铜有点凶万古尸王开海傅总,你老婆又娇又狠诸天大航海时代我能升级避难所奔驰吧足球骑马与萝莉斗罗里的藤虎一笑和死对头甜甜恋爱[快穿]魔兽使徒宋北云