郭長征一驚,轉頭看向窗外。
雖然此時沒有陽光直射進來,但外面的天色已經明顯泛白。
「這……居然快天亮了?」郭長征覺得一陣恍惚。
兩人這大半個晚上,連水都沒怎麼顧得上喝。
完全沉浸在那種發現新大陸的極度亢奮之中,根本沒有察覺到時間的流逝。
「效率太高了。」郭長征看著滿屏整潔的代碼。
「如果按照我以前的計劃,把這個複雜的網絡架構用底層代碼實現,起碼得兩個月。」
「而現在,一晚上,全搞定了。」
楚一航摸了摸肚子,感覺胃裡空空如也,餓得有點難受。
「郭老師,模型搭好了,咱們現在就讓它跑起來吧?」
「跑!必須跑!」郭長征此刻也顧不上疲憊。
楚一航打開瀏覽器,登錄了ImageNet的官方網站。
由於完整的數據集高達上千萬張圖片,龐大,全部下載不僅耗時,楚一航硬碟也裝不下。
於是,他們只勾選了其中一個非常小的數據子集,大概幾萬張圖片,涵蓋了貓、狗、汽車等常見的十幾個分類。
點擊下載。
趁著下載的間隙,楚一航寫了一個簡單的數據預處理腳本,把圖片統一縮放並轉化為框架能讀懂的張量格式。
下載完成,數據加載完畢。
楚一航深吸了一口氣,敲出了運行命令,按下回車。
工作站機箱裡的幾塊頂配AMD顯卡風扇,瞬間開始加速旋轉,發出低沉的轟鳴聲。
屏幕的終端上,開始一行一行地跳出訓練進度的提示。
進度條開始緩慢地向前推移。
郭長征和楚一航盯著那個進度條看了足足五分鐘。
「一航,這速度,大概要跑多久?」郭長征估算了一下時間。
楚一航看著系統給出的剩餘時間預估。
「郭老師,雖然咱們這是最頂配的顯卡,算力已經很強了。」
「但咱們這個網絡層數不少,哪怕只是幾萬張圖片的子集,跑完這50輪訓練,預估也得好幾個小時。」
此時,通宵過後的後遺症開始瘋狂反撲。
兩人都感覺到了那種又累、又困、又餓的虛脫感。
大腦已經開始有點不轉了。
楚一航站起身,活動了一下僵硬的關節。
「郭老師,你這也累壞了,再熬下去身體吃不消的。」
「我這房子正好是兩居室,咱正好一人一間房。」
楚一航指了指走廊。
「咱們先去休息一下,踏踏實實睡一覺。」
「機器就在這兒自己跑著,等結果出來了咱們再回來看。」
郭長征在腦子裡過了一下日期。
今天是周六,周末不用去學校上課,也沒有行政會議。
如果不看到最後的結果,他覺得自己就算回家也根本無法安心。
「行,一航,那我就厚著臉皮在你這兒叨擾一下了。」郭長徵答應下來。
兩人各自進了一間臥室,關上門。
楚一航連衣服都沒脫,倒在床上就閉上了眼睛。
但他發現,自己的大腦皮層極度活躍。
一閉上眼,滿腦子全是代碼、張量、矩陣和各種報錯信息。
他翻來覆去地換了好幾個姿勢,好不容易才迷迷糊糊地陷入了淺睡眠。
大概過了三個多小時。
楚一航猛地睜開眼睛,看了一眼手機上的時間。
上午九點十分。
他覺得自己好像睡了,又好像根本沒睡。
那種對實驗結果的強烈期待,就像一個小鉤子一樣,不斷地撓著他的神經。
他翻身下床,穿上拖鞋,走出臥室。
剛走到客廳,楚一航就愣住了。
郭長征已經坐在了工作站前面,眼睛死死地盯著屏幕。
「郭老師?」楚一航走過去打了個招呼。
「你這……怎麼就睡了這麼一會兒?不多休息一下?」
郭長征回過頭,眼睛裡布滿了紅血絲,但精神狀態卻異常亢奮。
「一航,我根本就沒睡著。」郭長征指了指自己。
「我在床上躺了兩個小時,腦子裡全在推演這個模型的結果。」
「因為太期待了,實在躺不住,就起來盯著了。」
楚一航拉開椅子坐下,看向屏幕。
終端上的提示顯示,訓練已經進行到了最後一輪的最後幾分鐘。
機箱風扇的轉速依舊很高。
兩人誰都沒有說話,客廳里安靜,只有機器的運轉聲。
終於,進度條走到了100%。
接著,程序開始在驗證集上進行最終的測試評估。
幾秒鐘後,兩行關鍵的評估數據列印在了屏幕上。
Top-1準確率: 50.23%。
Top-5準確率: 70.81%。
郭長征愣了幾秒,隨後猛地從椅子上站了起來。
楚一航也是一臉震驚,直接爆了句粗口:「臥槽!牛逼!」
兩人興奮地在半空中重重地擊了個掌。
兩聲清脆的巴掌聲在客廳里迴蕩。
他們太清楚這兩行數字所代表的含金量了。
激動得臉都紅了。
在普通人眼裡,這只是兩個乾巴巴的百分比。
但在懂行的科研人員眼裡,這就是炸裂的神跡。
所謂的Top-1準確率達到50%,究竟是個什麼概念?
簡單來說,隨便給這個剛搭好幾小時的AI模型扔一張它從來沒見過的圖片,比如一張貓的照片。
模型經過計算後給出的第一個判定答案,有足足一半的概率,會精準無誤地指出這是一隻「貓」。
雖然它也有一半的概率會猜成狗或者別的什麼東西,但這個首選命中率,在當時的技術背景下已經高得嚇人。
而Top-5準確率高達70%,則更加恐怖。
這意味著,哪怕模型偶爾犯迷糊,沒把正確答案排在頭名。
只要它給出的前五個最有可能的猜測名單里,比如依次寫著「老虎」、「小狗」、「貓」、「狐狸」、「越野車」。
由於「貓」這個正確選項出現在了前五名的列表里,在測試標準中就算它預測成功。
這種極具包容度的準確率,達到了驚人的百分之七十。
兩人之所以激動得有些失態,甚至在大清早的客廳里大喊大叫,是因為這個成績放在當下,完全是不可思議的。
要知道,他們今晚使用的資源極其寒酸。
為了圖快,他們根本沒有去下載ImageNet那多達一千五百萬張圖片的完整數據集。