《带着手机重生，目标科技教父》 第623章 这框架简直是神器！

楚一航负责敲代码，郭长征负责在旁边提供理论指导和架构设计。



“这里加一层池化层，文档上说这能降低数据维度，防止过拟合。”郭长征指挥着。



“好，步长设为2，卷积核大小设为3。”楚一航手指翻飞。



“这一层的激活函数换一下，别用原来的了，文档里推荐了一个叫relu的函数。”



“relu？我找找……有了，一行代码搞定。”



在搭建的过程中，他们也不是一帆风顺。



时不时会遇到一些报错。



比如“张量维度不匹配”这种经典的错误。



但得益于框架清晰的报错提示，他们总能迅速定位到是哪一层的矩阵乘法出了问题。



两个人为了算清楚上一层的输出怎么和下一层的输入对齐，甚至找了张a4纸在桌子上画起了矩阵图。



不知不觉中，时间过得飞快。



楚一航敲击回车，最后一行代码编写完毕。



他们把整个神经网络的模型类封装好了，并且配置好了损失函数和优化器。



楚一航伸了个大大的懒腰，只觉得脖子和后背一阵酸痛。



郭长征也揉了揉酸涩的眼睛。



窗外传来了鸟叫声。



楚一航听到后愣了一下，瞥了一眼屏幕右下角的时间，整个人愣住了。



“郭老师……五点半了。”



『加入书签，方便阅读』(3/3)