文章/答案/技术大牛

发布

社区首页 >问答首页 >ai.djl.engine.EngineException: PytorchStreamReader定位文件失败constants.pkl:文件未找到

问ai.djl.engine.EngineException: PytorchStreamReader定位文件失败constants.pkl:文件未找到
EN

Stack Overflow用户

提问于 2022-01-27 01:03:49

回答 1查看 470关注 0票数 0

我试图使用DJL加载Yolov5模型以进行目标识别，但报告了一个错误。

ai.djl.engine.EngineException: PytorchStreamReader failed locating file constants.pkl: file not found
        at ai.djl.pytorch.jni.PyTorchLibrary.moduleLoad(Native Method)
        at ai.djl.pytorch.jni.JniUtils.loadModule(JniUtils.java:1360)
        at ai.djl.pytorch.engine.PtModel.load(PtModel.java:89)
        at ai.djl.repository.zoo.BaseModelLoader.loadModel(BaseModelLoader.java:156)
        at ai.djl.repository.zoo.Criteria.loadModel(Criteria.java:166)
        at ai.djl.repository.zoo.ModelZoo.loadModel(ModelZoo.java:132)
        at com.czworks.dji.Test10.main(Test10.java:58)

pytorch

djl

回答 1

Stack Overflow用户

发布于 2022-04-22 15:59:56

DJL使用PyTorch C++ API，它只支持TorchScript模型。你需要追踪你的模型。

请参阅以下文档：如何将PyTorch模型转换为TorchScript模型

TorchScript模型在C++中的应用

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/70872112

复制

相似问题

问ai.djl.engine.EngineException: PytorchStreamReader定位文件失败constants.pkl:文件未找到
EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问ai.djl.engine.EngineException: PytorchStreamReader定位文件失败constants.pkl:文件未找到EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问ai.djl.engine.EngineException: PytorchStreamReader定位文件失败constants.pkl:文件未找到
EN