两人平时研究的都是些传统的机器学习算法,对这种结构并不算特别清楚。
直到看到文档里清清楚楚地写着:该模块通过卷积核提取局部特征,极为适用于图像识别与计算机视觉领域。
“图像识别?”郭长征敏锐地抓住了这个词。
“我们下午吃饭的时候,讨论的李飞飞教授那个Image数据集,不就是图像分类吗?”
楚一航一拍桌子:“绝了!连这种专门针对图像的模块都提前封装好了。”
“这简直就是给咱们量身定制的武器库啊。”
两人对着这些功能模块,越讨论越兴奋。
“一航,别光看着眼馋了,纸上得来终觉浅。”郭长征提议。
“咱们现在就动手,按这个文档,写一个简单的几层网络试试水。”
“好嘞。”楚一航立刻切换到代码编辑器。
按照平时用CUDA编写底层算力代码的习惯,楚一航习惯性地准备先写一堆内存分配的语句。
The content is not finished, continue reading on the next page