为什么Keras和Pytorch都是把模型保存成h5的格式而不是pkl格式?有特别的原因吗?
2个回答
可以保存为pkl格式的,你可以在Pytorch中这样试试:
#保留全部的神经网络
torch.save(net1 ,'net.pkl') # net1 是你要保存的网络
net2 = torch.load('.//pkl//net.pkl') # 加载你的网络
#只保留神经网络的参数
torch.save(net1.state_dict(), 'net_params.pkl' )
# 用参数还原神经网络 !!首先我们必须创造一个和原来具有一样结构的神经网络
net2 = torch.load('.//pkl//net.pkl')
http://sofasofa.io/tutorials/pytorch_tutorials/ 这里有一篇Pytorch的简单教程
SofaSofa数据科学社区DS面试题库 DS面经
pkl太大了,一般存成h5吧
-
abuu
2019-08-29 14:06