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

  统计/机器学习 Python TensorFlow    浏览次数:1546        分享
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模型后怎么画出loss和val_loss的对比图?   1回答

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

keras load_model报错   1回答

keras和tensorflow有什么关系?   3回答

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

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

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

keras.models.Sequential()是一种什么模型   1回答

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

keras hdf5 從某層輸入數值   1回答

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

keras模型合并请教   2回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!