DL4J中保存的多层网络未在Python中加载

2024-09-29 00:13:43 发布

您现在位置:Python中文网/ 问答频道 /正文

我一直在Deeplearning4j中使用神经网络,需要将其转换为Python。为了使用相同的模型(DL4J中的多层网络),我将其保存为.h5文件。像这样:

File newFile = new File("newModel.h5");
ModelSerializer.writeModel(network, newFile, true);

现在,当我尝试在Python中加载它时,会出现以下错误:

OSError: SavedModel file does not exist at: newModel.h5/{saved_model.pbtxt|saved_model.pb}

我尝试过使用不同的扩展,比如.pb,并在python中使用了相对路径和绝对路径。没有任何帮助。有人能解释一下为什么会发生这种情况吗?互联网上似乎没有足够的关于这个问题的信息,用python实现相同代码的唯一方法似乎是训练一个新模型,等等


Tags: 文件模型网络newmodel神经网络fileh5
1条回答
网友
1楼 · 发布于 2024-09-29 00:13:43

dl4j模型是一个zip文件。你能澄清一下你想做什么吗?如果从keras导入并需要重新保存,最好将权重导出为numpy数组并重新创建architecutre。您可以使用model.params()来实现这一点,它为您提供了权重

相关问题 更多 >