用DJL加载PyTorch模型的路径

Question

我训练了一个定制的PyTorch模型，并将它保存为一个.pt文件。我现在正尝试使用DJL在Java中加载这个模型。

Path modelDir = Paths.get("/Users/myname/eclipse-workspace/myProject/src/ML/");
Model model = Model.newInstance("model.pt");
model.load(modelDir);

然而，这就产生了以下例外情况：

ai.djl.engine.EngineException: No deep learning engine found. --我发现，即使我将路径更改为完全无效的东西，我也会得到相同的错误。因此，我认为问题在于路径或模型名称。我做错了什么？顺便说一下，我正在通过Eclipse中的Maven运行这个项目。

谢谢!

Answer 1

从错误消息来看，您似乎还没有在项目中指定PyTorch引擎/本机依赖项。你得先提供这个。然后，请按照下面的说明：model.html加载您的py手电筒模型

Answer 2

您需要在项目中包括喷炬发动机组件：

<dependency>
    <groupId>ai.djl.pytorch</groupId>
    <artifactId>pytorch-engine</artifactId>
    <version>0.16.0</version>
    <scope>runtime</scope>
</dependency>

DJL支持多个引擎(例如PyTorch、TensorFlow、Apache MXNet)。这些引擎包是在单独的maven包中提供的。DJL将在运行时在类路径中找到引擎。