Skip to content

Blog

Direct Preference Optimization (DPO) 訓練方法筆記

介紹

DPODirect Preference Optimization, 直接偏好優化)是一種取代 RLHFReinforcement Learning from Human Feedback, 基於人類反饋的強化學習)的微調方式。眾所皆知,大型語言模型在經過非監督式學習後能夠學習到大量的知識與理解能力(有些研究者認為是『壓縮並保存』了知識在神經網路權重中);在監督式學習後學會了流暢地回應我們的問題,或者說是學會了『對話』的能力。

Read More »Direct Preference Optimization (DPO) 訓練方法筆記

[GitHub] 發 Pull Request (PR) 的流程筆記

介紹

在 GitHub 上發 Pull Request (PR) 給開源項目是一件美好卻又重要的事情。簡單來說,PR 是一個當你看到開源專案時,發現有個真的很想做的功能/很想修復的 BUG,你可以嘗試將該開源專案分岔(fork)成自己的私有專案,並在上面進行開發;開發完成後,再藉由提交 PR 給原始的專案維護者做合併檢查,看是否要合併回原始專案的主分支。

Read More »[GitHub] 發 Pull Request (PR) 的流程筆記

[PyTorch] 將模型刪除後,釋放 GPU / CPU 的記憶體空間

問題描述

昨晚,我在改進一段將模型融合的程式碼時,由於個人設備資源不足,我採取分層合併的方法,一次只取一層的記憶體進行合併,以此減少系統同時保存的模型記憶體大小。然而,我發現模型雖然在 GPU 中的記憶體很容易被刪除、但是 CPU 中的記憶體被回收是十分不容易的。這涉及到了 Python 的資源回收器的設計。

Read More »[PyTorch] 將模型刪除後,釋放 GPU / CPU 的記憶體空間