Skip to content

3 月 2022

[PyTorch] 保存優化器(optimizer)來接續訓練模型

Last Updated on 2022-03-15 by Clay

我一直以來都希望能夠保存 PyTorch 訓練模型時所使用的優化器optimizer),以便能夠在模型結束訓練之後,繼續往下訓練;一般來說,如果是要做遷移學習、微調模型(fine-tune),那麼並不需要特別將上一次訓練的優化器保存下來。

Read More »[PyTorch] 保存優化器(optimizer)來接續訓練模型
Exit mobile version