torchline 讓Pytorch使用的更加順滑

2022-03-30 01:43:19 字數 874 閱讀 4775

相信大家平時在使用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那麼讓人普及,畢竟是要收費...