热天中文网

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

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

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

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

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

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

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

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

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

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

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

“图像识別?”

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

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

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

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

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

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

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

郭长征提议。

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

“好嘞。”

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

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

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

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

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

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

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

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

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

不到二十行代码。

简洁,优雅。

“这就写完了?”

楚一航自己都不敢相信。

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

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

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

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

新书推荐

影视诸天:从正阳门下开始模拟神仙是什么体验夫人别嫁了,主帅他不孕不育啊等风的女孩快穿:炮灰女配不走心签到天师宫,下山已无敌数风流人物乖宝别哭!禁欲大佬掐腰哄惊悚狩猎计划[无限]快穿之每个世界秀恩爱这个火影不热血帝妃她富敌三界君临彼岸无限版帝国时代一念成仙我成了山海界灵英雄联盟,云起瓦罗兰漫威世界的武神无上崛起您的无限生存游戏已开启雪鹰领主反派穿成Omega以后他A爆了.总裁爹地,不许欺负我妈咪!美漫世界的巫妖王凶狠狠的指挥官,晚上粘着要亲亲