自從我學習了Deep Learning後,每天都跟學長學習怎麼train一個好的model。學長每天晚上都會到我房間來指導我,可能是因為我的電腦有好一點的顯卡,train起來比較順比較快。這天學長又來到我房間指導我…
「學弟,我們今天要來train一個比較大的資料集…」
「這不是ImageNet嗎?學長這個太大了,顯卡會吃不進去的…」
這時學長又打開他已經架構好的網路,說:「我們今天用特別一點,transfer learning。」
他一打開他那網路,我驚呼著上下打量:「啊,這是VGG19,啊…不行,學長,這個網路真的太大了,會壞掉的。」
學長看到我那驚恐的表情,臉上些許得意,他撫摩著滑鼠對我邪笑:「學弟不要怕,我相信你的顯卡可以吃進去的。」
還沒等我反應過來,學長一口氣用力地enter +shift下去,整個資料集model.fit進來。電腦被這促急不防的攻勢嚇到,風扇開始瘋狂運轉,我則是已經尖叫了出來:「學長,你怎麼就放進去了!會壞掉的!」
學長不顧我的反應,開始看著每一個epoch那塞好塞滿的資料集跑完,跑完一個epoch後,資料集出來shuffle後一遍又進去model train。
我的顯卡受不了他這麼搞,我抓住學長的手腕說:「學長不行了,至少…至少…讓我們train on batch…不然真的要吃不下了啊啊!」
「不行!我今天就是來測試你的顯卡能耐如何,壞掉了又怎麼樣,學長…學長再買一個給你,呃!」語畢,資料集跑完一個epoch,又用力地塞回網路。
「啊!學長!你…你好壞!」
這時我發現,validation的loss,突然上升,我趕緊跟學長說:「學長這是不是要overfitting了…」伴隨著風扇願轉越大聲,我心裡開始不安了起來。
「老師上次開會不是就說想看你overfitting的樣子嗎?」學長完全沒想停手的意思,眼神近乎瘋狂。
「不行!不行!不能overfitting,啊啊!風扇轉好快!顯卡…顯卡…要壞掉了啊啊啊啊。」
#靠交26