将 ComputationGraph 放入 JAR DL4j 的资源目录中

Putting a ComputationGraph in the resources directory in a JAR DL4j

在正常情况下,我可以将一个 ComputationGraph(一个卷积神经网络)保存到一个文件中,稍后再加载它 运行,它工作正常。

但是,当我将它包含在一个 jar 中并尝试加载它时,它失败了。

有什么方法可以从 Jar 内部加载 ComputationGraph 对象作为资源吗?

我让它适用于 Word2Vec 对象,但它在 ComputationGraph 对象上失败了。

谢谢

它只是一个 zip 文件(或者应该是!)。我不确定您是否正在使用我们的 ModelSerializer(这是您唯一应该用于计算图或多层网络的东西) - 您可以在调用时使用 InputStream 或 File:

ModelServer.restoreComputationGraph

在 InputStream 上。 InputStream 可以是类路径资源输入流(使用类似 spring 的 ClassPathResource 的东西)或调用 YourClass.class.getResourceAsStream(..) - 任何一个都应该适合你。