新聞中心
這里有您想知道的互聯(lián)網(wǎng)營銷解決方案
Pytorch之保存讀取模型實例-創(chuàng)新互聯(lián)
pytorch保存數(shù)據(jù)
pytorch保存數(shù)據(jù)的格式為.t7文件或者.pth文件,t7文件是沿用torch7中讀取模型權(quán)重的方式。而pth文件是python中存儲文件的常用格式。而在keras中則是使用.h6文件。
# 保存模型示例代碼 print('===> Saving models...') state = { 'state': model.state_dict(), 'epoch': epoch # 將epoch一并保存 } if not os.path.isdir('checkpoint'): os.mkdir('checkpoint') torch.save(state, './checkpoint/autoencoder.t7')
另外有需要云服務(wù)器可以了解下創(chuàng)新互聯(lián)scvps.cn,海內(nèi)外云服務(wù)器15元起步,三天無理由+7*72小時售后在線,公司持有idc許可證,提供“云服務(wù)器、裸金屬服務(wù)器、高防服務(wù)器、香港服務(wù)器、美國服務(wù)器、虛擬主機、免備案服務(wù)器”等云主機租用服務(wù)以及企業(yè)上云的綜合解決方案,具有“安全穩(wěn)定、簡單易用、服務(wù)可用性高、性價比高”等特點與優(yōu)勢,專為企業(yè)上云打造定制,能夠滿足用戶豐富、多元化的應(yīng)用場景需求。
本文標題:Pytorch之保存讀取模型實例-創(chuàng)新互聯(lián)
文章路徑:http://fisionsoft.com.cn/article/djghcp.html