Keras问题:Save Model: TypeError: ('Not JSON Serializable:',..乱码 )

简介

今天在修改模型之后,用keras继续跑模型的时候,发现模型无法保存,在checkpoint的地方,后面在fit函数里面的callback中一直报错,出现的错误可简单理解为模型无法序列化到json文件中,对于该问题尝试了很多种方法,包含这里的方法,但是我发现在,tensorflow下面的Sequence.py下面是有这些的,我用的是tensorflow1.14。

解决方法

在我试了很多方法之后,我最后发现,在Checkpoint中,修改保存整个模型,转而只保存权值之后,可以对模型进行保存,从而继续进行训练

mc = ModelCheckpoint('../Models/BrainAgeSFCN_SSE_8_90',verbose=1,mode='min',save_best_only=True,save_weights_only=True)

再次更新,这次找到了问题所在,问题是keras中,如果存在自定层,例如lamda层,则用checkpoint就不能保存模型,需要只保存权值,模型在需要预测的时候在重新加载。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容