为什么Keras和Pytorch都是把模型保存成h5的格式而不是pkl格式?

  统计/机器学习 Python TensorFlow I/O    浏览次数:6708        分享
0

为什么Keras和Pytorch都是把模型保存成h5的格式而不是pkl格式?有特别的原因吗?

 

thatdeep   2019-08-18 14:31



   2个回答 
5

不建议把神经网络模型或者数据集存成pkl格式,因为太占用空间了。h5更小。

在h5中100mb的模型,存到pkl里要300mb以上。

SofaSofa数据科学社区DS面试题库 DS面经

strong.man   2019-08-28 09:39

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面经

liuyang   2019-08-19 09:13

pkl太大了,一般存成h5吧 - abuu   2019-08-29 14:06


  相关讨论

Keras模型针对同一数据集做fit和evaluation,accuracy值却不相同

keras无法加载imdb文本:Object arrays cannot be loaded when allow_pickle=False

keras load_model报错

keras和tensorflow有什么关系?

keras加载imdb数据时num_words参数是干嘛的?

keras sequential model怎么增加一个dropout层?

训练完keras模型后怎么画出loss和val_loss的对比图?

keras里的predict,predict_classes,predict_proba有什么区别?

怎么在keras中加载MNSIT数据集?

keras怎么设置/改动rmsprop的学习率?

  随便看看

为什么LASSO可以做特征选择,而Ridge却不行?

主成分分析法(PCA)算是黑盒算法吗?

tsne被忽视?tsne有什么缺点?

python里怎么表示科学计数法?

python产生服从常用概率分布的随机数