如何用Colab訓練機器學習模型?
介紹
大家好,今天想跟大家分享的是如何在colab訓練機器學習模型,會採用colab訓練,主要是由於有時候自己電腦的性能有限,colab有提供免費也強大的運算資源可以使用。在訓練時間沒有長達到好幾天的情形,用colab訓練是很方便的一件事情。
以下就簡單分享是如何使用colab進行訓練。
實作教學
新增與撰寫筆記本
一開始可搜尋colab點擊進去。
![](https://i0.wp.com/peterlihouse.com/wp-content/uploads/2025/01/image.png?resize=1024%2C481&ssl=1)
點擊進去後,會出現以下畫面,他會顯示最近使用的筆記本,所以如果要使用上次的筆記本直接在中間的方框點擊就好,那如果還沒有建立的筆記本,就可以點新增筆記本。
![](https://i0.wp.com/peterlihouse.com/wp-content/uploads/2025/01/image-1.png?resize=1024%2C480&ssl=1)
點擊完後,會出現新的筆記本可以在上面輸入程式碼。
![](https://i0.wp.com/peterlihouse.com/wp-content/uploads/2025/01/image-2.png?resize=1024%2C468&ssl=1)
這邊就直接輸入機器學習訓練的程式碼,是用CNN訓練cifar10的模型。在這一步程式碼已經建立完成後,要開始配置運算資源。
![](https://i0.wp.com/peterlihouse.com/wp-content/uploads/2025/01/image-3.png?resize=1024%2C472&ssl=1)
配置執行資源
點擊上方欄位,會有執行階段,點入後,可點擊變更執行階段類型。
![](https://i0.wp.com/peterlihouse.com/wp-content/uploads/2025/01/image-4.png?resize=1024%2C488&ssl=1)
就可以看到有很多運算資源的選擇,有CPU,GPU,TPU。對於要訓練CNN這種圖像辨識比較複雜的模型,會採用GPU來進行運算。
![](https://i0.wp.com/peterlihouse.com/wp-content/uploads/2025/01/image-5.png?resize=1024%2C480&ssl=1)
所以我們就點選GPU的選項,再按下儲存作確定。
![](https://i0.wp.com/peterlihouse.com/wp-content/uploads/2025/01/image-6.png?resize=1024%2C484&ssl=1)
執行機器學習訓練
程式碼輸入完後,就可點擊全部執行,就會將全部程式碼執行完畢。
![](https://i0.wp.com/peterlihouse.com/wp-content/uploads/2025/01/image-7.png?resize=1024%2C484&ssl=1)
這時候可以把畫面往下滑,滑到有顯示執行結果的部分,像是我們這邊一開始會先下載cifar10的資料,以及有不同epoch的訓練結果。會有精確度與損失。
![](https://i0.wp.com/peterlihouse.com/wp-content/uploads/2025/01/image-8.png?resize=1024%2C484&ssl=1)
訓練完成後,終止工作階段
由於使用GPU的運算資源,運算成本是相當昂貴的,colab針對GPU會有一定執行限制,所以如果在閒置狀態,沒有使用,就要把GPU運算資源取消。
在訓練完成後,我們就點擊執行階段,再點擊管理工作階段。
![](https://i0.wp.com/peterlihouse.com/wp-content/uploads/2025/01/image-9.png?resize=1024%2C484&ssl=1)
可以看到我們還在執行中的筆記本,這時候可點擊最右邊,垃圾桶刪除符號。
![](https://i0.wp.com/peterlihouse.com/wp-content/uploads/2025/01/image-10.png?resize=1024%2C480&ssl=1)
就會出現終止目前工作階段,就點擊終止。
![](https://i0.wp.com/peterlihouse.com/wp-content/uploads/2025/01/image-11.png?resize=1024%2C480&ssl=1)
原本在畫面上的執行階段就會消失,代表我們就把GPU資源釋放出來。這樣就結束,可以關掉筆記本了。
![](https://i0.wp.com/peterlihouse.com/wp-content/uploads/2025/01/image-12.png?resize=1024%2C480&ssl=1)
結論
本篇文章簡單跟大家分享一下colab訓練的操作教學,用colab訓練好處是完全使用雲端上的資源,自己的電腦就不用特別花資源去做計算。