两人彻底进入了忘我的状态。
楚一航负责敲代码,郭长征负责在旁边提供理论指导和架构设计。
“这里加一层池化层,文档上说这能降低数据维度,防止过拟合。”郭长征指挥着。
“好,步长设为2,卷积核大小设为3。”楚一航手指翻飞。
“这一层的激活函数换一下,别用原来的了,文档里推荐了一个叫ReLU的函数。”
“ReLU?我找找……有了,一行代码搞定。”
在搭建的过程中,他们也不是一帆风顺。
时不时会遇到一些报错。
比如“张量维度不匹配”这种经典的错误。
但得益于框架清晰的报错提示,他们总能迅速定位到是哪一层的矩阵乘法出了问题。
The content is not finished, continue reading on the next page