java - 将 ComputationGraph 放在 JAR DL4j 的资源目录中

Question

在正常情况下，我可以将 ComputationGraph（卷积神经网络）保存到文件中并在以后的运行中加载它，它工作正常。

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

有没有办法从 Jar 内部加载 ComputationGraph 对象作为资源？

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

谢谢

Answer 1

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

ModelServer.restoreComputationGraph

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