醉书楼 > 都市小说 > 带着手机重生,目标科技教父TXT下载 > 带着手机重生,目标科技教父最新章节列表 > 第623章 这框架简直是神器!(第2节)

带着手机重生,目标科技教父最新章节全文阅读

作者:没钱又任性  加入书架  带着手机重生,目标科技教父没钱又任性  带着手机重生,目标科技教父全文阅读
带着手机重生,目标科技教父最新章节第800章 马老师约饭(26-06-07)    第799章 要快(26-06-07)    第798章 刚甩律师函,转头就要写道歉信(26-06-07)    

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

苦逼的算法研究员来说,简直就是久旱逢甘霖。

他们觉得之前只知道盛夏科技批了算力,居然没静下心来仔细研究这个框架,简直是暴殄天物。

“继续往下看,看看还有什么惊喜。”郭长征现在完全没有了困意。

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

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

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

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

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

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

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

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

“图像识别?”郭长征敏锐地抓住了这个词。

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

楚一航一拍桌子:“绝了!连这种专门针对图像的模块都提前封装好了。”

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

两人对着这些功能模块,越讨论越兴奋。

“一航,别光看着眼馋了,纸上得来终觉浅。”郭长征提议。

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

“好嘞。”楚一航立刻切换到代码编辑器。

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

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

程序员必须手动在主机的内存里开辟空间,再在显卡的显存里开辟空间。

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

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

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

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

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

不到二十行代码。

简洁,优雅。

“这就写完了?”楚一航自己都不敢相信。

本章未完,请点击"下一页"继续阅读! 第2页 / 共3页