【Pytorch】.pth權重文件解析

pytorch最後的權重文件是.pth格式的。 經常遇到的問題： 進行finutune時，改配置文件中的學習率，發現程序跑起來後竟然保持了以前的學習率， 並沒有使用新的學習率。 原因： 首先查看.pth文件中的內容，我們發現它其實是一個字典格式的文件 其中保存了optimizer和scheduler，所以再次加載此文件時會使用之前的學習率。 我們只需要權重，也就是model部分，將其導出就可以了 

>>阅读原文<<