|

如何用Colab訓練機器學習模型?

介紹

大家好,今天想跟大家分享的是如何在colab訓練機器學習模型,會採用colab訓練,主要是由於有時候自己電腦的性能有限,colab有提供免費也強大的運算資源可以使用。在訓練時間沒有長達到好幾天的情形,用colab訓練是很方便的一件事情。

以下就簡單分享是如何使用colab進行訓練。

實作教學

新增與撰寫筆記本

一開始可搜尋colab點擊進去。

點擊進去後,會出現以下畫面,他會顯示最近使用的筆記本,所以如果要使用上次的筆記本直接在中間的方框點擊就好,那如果還沒有建立的筆記本,就可以點新增筆記本。

點擊完後,會出現新的筆記本可以在上面輸入程式碼。

這邊就直接輸入機器學習訓練的程式碼,是用CNN訓練cifar10的模型。在這一步程式碼已經建立完成後,要開始配置運算資源。

配置執行資源

點擊上方欄位,會有執行階段,點入後,可點擊變更執行階段類型。

就可以看到有很多運算資源的選擇,有CPU,GPU,TPU。對於要訓練CNN這種圖像辨識比較複雜的模型,會採用GPU來進行運算。

所以我們就點選GPU的選項,再按下儲存作確定。

執行機器學習訓練

程式碼輸入完後,就可點擊全部執行,就會將全部程式碼執行完畢。

這時候可以把畫面往下滑,滑到有顯示執行結果的部分,像是我們這邊一開始會先下載cifar10的資料,以及有不同epoch的訓練結果。會有精確度與損失。

訓練完成後,終止工作階段

由於使用GPU的運算資源,運算成本是相當昂貴的,colab針對GPU會有一定執行限制,所以如果在閒置狀態,沒有使用,就要把GPU運算資源取消。

在訓練完成後,我們就點擊執行階段,再點擊管理工作階段。

可以看到我們還在執行中的筆記本,這時候可點擊最右邊,垃圾桶刪除符號。

就會出現終止目前工作階段,就點擊終止。

原本在畫面上的執行階段就會消失,代表我們就把GPU資源釋放出來。這樣就結束,可以關掉筆記本了。

結論

本篇文章簡單跟大家分享一下colab訓練的操作教學,用colab訓練好處是完全使用雲端上的資源,自己的電腦就不用特別花資源去做計算。

Similar Posts

發佈留言

發佈留言必須填寫的電子郵件地址不會公開。 必填欄位標示為 *