第623章這框架簡直是神器！
第623章這框架簡直是神器！(第3/3頁)
這個，咱們今晚完全可以把吃飯時構思的那個真正的網絡架構給搭出來。”郭長征提議。
“乾！”楚一航乾勁十足。
兩人徹底進入了忘我的狀態。
楚一航負責敲代碼，郭長征負責在旁邊提供理論指導和架構設計。
“這裡加一層池化層，文檔上說這能降低數據維度，防止過擬合。”郭長征指揮著。
“好，步長設為2，卷積核大小設為3。”楚一航手指翻飛。
“這一層的激活函數換一下，彆用原來的了，文檔裡推薦了一個叫relu的函數。”
“relu？我找找……有了，一行代碼搞定。”
在搭建的過程中，他們也不是一帆風順。
時不時會遇到一些報錯。
比如“張量維度不匹配”這種經典的錯誤。
但得益於框架清晰的報錯提示，他們總能迅速定位到是哪一層的矩陣乘法出了問題。
兩個人為了算清楚上一層的輸出怎麼和下一層的輸入對齊，甚至找了張a4紙在桌子上畫起了矩陣圖。
不知不覺中，時間過得飛快。
楚一航敲擊回車，最後一行代碼編寫完畢。
他們把整個神經網絡的模型類封裝好了，並且配置好了損失函數和優化器。
楚一航伸了個大大的懶腰，隻覺得脖子和後背一陣酸痛。
郭長征也揉了揉酸澀的眼睛。
窗外傳來了鳥叫聲。
楚一航聽到後愣了一下，瞥了一眼屏幕右下角的時間，整個人愣住了。
“郭老師……五點半了。”