【文檔學習】PyTorch——模型的保存和加載

本系列記錄了博主學習PyTorch過程中的筆記,官方tutorial。 本文介紹了模型的保存和加載,官方示例。更新於2019.03.21。 這個文檔提供了關於保存和加載PyTorch模型的一系列使用案例。可以全看,也可以根據需要選擇。 在保存和加載模塊時,需要了解下面三個核心函數: torch.save:保存一個序列化(serialized)的目標到磁盤。函數使用了Python的pickle程序用
相關文章
相關標籤/搜索