相信大家平時在使用pytorch搭建網路時,多少還是會覺得繁瑣,因為我們需要搭建資料讀取,模型,訓練,checkpoints儲存等等一系列模組。每當切換到新的任務後很多情況下之前的**不能復用,或者說要復用就需要做很多地方的修改,到最後還不如重新寫一遍。所幸,pytorch_lightning讓這一過程簡化了很多,相信如果你用過這個庫你也會體驗到它的方便性。但是torchline的存在是讓你使用pytorch更加的順滑舒暢。
torchline基於pytorch_lightning (pl)開發,整個庫的結構設計借鑑了detectron2,具體可以閱讀下面幾篇文章進行了解:
如下圖所示,灰色部分pl可以自動完成。我們需要做的,差不多也就載入資料、定義模型、確定訓練和驗證過程
torchline則進一步簡化,而且可以讓你的模型復用性更高。
以構建模型為例進行大致的介紹(細節可以去github檢視),假如你之前建立了乙個mymodel
的模型,之後如果你想使用這個模型,你只需要在config檔案中將model.name
修改成mymodel
(即只是修改字串的值,之後torchline
會自動切換模型)。
總的來說,pytorch_lightning有的torchline肯定都有哈哈哈,但是使用起來**復用性和易用性更高,歡迎去github品嚐,覺得好用麻煩star,也歡迎issue討論。
如有意合作,歡迎私戳
2019-12-13 21:31:15
什麼是PyTorch,為何要使用PyTorch
pytorch 是torch7 團隊開發的,從它的名字就可以看出,其與torch 的不同之處在於pytorch 使用了python 作為開發語言。所謂 python first 同樣說明它是乙個以python 優先的深度學習框架,不僅能夠實現強大的gpu 加速,同時還支援動態神經網路,這是現在很多主...
從頭開始安裝python環境和pytorch等
參考資料 檢視cuda版本 將anaconda安裝到pycharm中 安裝pytorch 直接安裝anaconda,版本選擇 anaconda3 5.2.0 windows x86 64版本,按照網上教程按步驟安裝anaconda,並且在安裝完之後用在cmd下直接pip list,顯示已經安裝了的包...
cedega讓我快樂!讓痛苦!
在linux上玩遊戲早不是夢想,很早以前,前輩們都用wine來實現。而複雜的配置讓很對初學者望而卻步。現在,菜鳥也可以輕鬆的在linux上跑起來。cs,極品飛車,魔獸,魔獸世界,一大堆的遊戲都可以在輕易的在linux下跑。它就是cedega,其實它早就有了,只是沒有wine那麼讓人普及,畢竟是要收費...