分类：load_model

2020
10-08

keras的load_model实现加载含有参数的自定义模型

网上的教程大多数是教大家如何加载自定义模型和函数，如下图这个SelfAttention层是在训练过程自己定义的一个class，但如果要加载这个自定义层，需要在load_model里添加custom_objects字典，这个自定义的类，不要用import，最好是直接复制进再训练的模型中，这些是基本教程。------------------分割线讲重点------------------如果直接运行上面的代码，会出现一个init初始化错误，如下图，再来看看这个SelfAttention自定义的类的... 继续阅读 >

2020
10-07

Keras load_model 导入错误的解决方式

在使用Kerasload_model时，会出现以下报错：ImportError:Failedtoimportpydot.Youmustinstallpydotandgraphvizfor`pydotprint`towork.解决办法：$pipinstallpydot$sudoapt-getinstallgraphviz补充知识：Keras保存model到指定文件夹和加载load_model指定文件夹中的文件（踩坑）我们一般保存模型和加载模型都是通过下面这段代码：fromkeras.modelsimportload_modelmodel.save('model.h5')fromkeras.model... 继续阅读 >