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

Question

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

但是，当我将其放入罐子并尝试加载时，它会失败。

是否有某种方法可以从Jar内部加载ComputationGraph对象作为资源？

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

谢谢

Answer 1

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

ModelServer.restoreComputationGraph

在InputStream上。该InputStream可以是类路径资源输入流（使用诸如spring的ClassPathResource之类的东西）或调用YourClass.class.getResourceAsStream（..）-任何一个都应该对您有用。