如果想要學習深度學習,用pytorch,有必要閱讀pytorch原始碼嗎?
作者:由 斑馬 發表于 歷史時間:2019-08-27
之前實驗室有個專案需要把訓練好的模型存到資料庫中,但是Keras匯出模型只能導到硬碟,所以實驗室小夥伴把Keras模型匯出硬碟然後讀取硬碟模型,然後再存資料庫。我需要把模型從資料庫中讀出來,載入到記憶體中,感覺他這麼幹有點蠢,看了Keras原始碼,修改一下,直接在記憶體裡面操作不經過硬碟。
建議你先學習深度學習原理吧 跑幾個模型試試 原始碼先不著急
看你的應用目的,一般情況下,只需要呼叫python或者c++的API介面進行訓練或者推理就可以了
pytorch, tensorflow, mxnet內部的實現機制目前正趨近於同質化,對於想研究深度學習平臺方向的人來說,是需要看程式碼的(例如:深度學習平臺搭建,結合業務的改進,底層最佳化等)
如果想學的深的話可以閱讀原始碼,但是實際中我們一般只搭個網路,訓個模型就行了,一般不會讀原始碼的。現在深度學習框架的原始碼都是由一個團隊共同開發的,每個人只負責一塊,閱讀原始碼還是不太容易的
我暈。看了提問者在某個答案下的回覆才明白,提問者說的原始碼是用pytorch開發的程式碼樣例,不是實現pytorch的原始碼。照這麼說的話,您還是多看看“原始碼”吧,不看別人的程式碼,很難提高自己的水平。
最好還是要閱讀原始碼的能力。你可以不去看原始碼,但是在你想進行網路結構改進的時候就必須要具備這個能力。如果是初學者可以透過
https://
morvanzhou。github。io/
入門。然後學習經典論文,在github上找相對應的程式碼復現。然後就是更改來提高結果。如果有需要用GPU的朋友可以參考一下公眾號: 智星AI,python pytorch都是配置好了的。可以直接執行程式。