1. 程式人生 > 實用技巧 >Pytorch中利用GPU訓練,CPU測試

Pytorch中利用GPU訓練,CPU測試

要先利用GPU訓練,CPU測試,那麼在模型訓練時候,是能儲存模型的引數而不能儲存整個模型,可見Pytorch模型儲存機制便可以學會模型的儲存、載入、測試