pytorch 模型加載與保存

翻譯自https://pytorch.org/tutorials/beginner/saving_loading_models.html 保存和與加載模型,有三個核心函數需要熟悉: torch.save: 保存一個序列化的對象至硬盤,。該函數使用了Python的pickle包用於序列化。模型、張量和各種對象的字典都可以使用該函數保存; torch.load:使用pickle的反序列化功能將序列化的
相關文章
相關標籤/搜索