如何在DJL中调用paddle语言处理模型
收藏
【问题描述】
想把paddle语言处理模型通过DJL加载到JAVA应用中调用,目前模型可以加载进去,但是在调用时报错;没有看到合适的入参处理案例,DJL官网的案例是图片识别的;测试用的模型是官网的 lac 模型;
Criteria criteria = Criteria.builder()
.setTypes(String.class, String.class)
.optModelPath(Paths.get("D:\\djl\\lac\\lac.zip"))
.optTranslator(new PaddleTranslator())
.optModelName("lac")
.build();
报错如下:
Exception in thread "main" ai.djl.translate.TranslateException: java.lang.UnsupportedOperationException: This NDArray implementation does not currently support this operation
at ai.djl.inference.Predictor.batchPredict(Predictor.java:189)
at ai.djl.inference.Predictor.predict(Predictor.java:126)
at djl.HuggingfaceTest.loadPaddle(HuggingfaceTest.java:145)
at djl.HuggingfaceTest.main(HuggingfaceTest.java:62)
Caused by: java.lang.UnsupportedOperationException: This NDArray implementation does not currently support this operation
at ai.djl.ndarray.NDArrayAdapter.getAlternativeArray(NDArrayAdapter.java:1225)
at ai.djl.ndarray.NDArrayAdapter.getNDArrayInternal(NDArrayAdapter.java:1173)
at ai.djl.ndarray.NDArrays.stack(NDArrays.java:1825)
at ai.djl.ndarray.NDArrays.stack(NDArrays.java:1785)
at ai.djl.translate.StackBatchifier.batchify(StackBatchifier.java:52)
at ai.djl.inference.Predictor.processInputs(Predictor.java:217)
at ai.djl.inference.Predictor.batchPredict(Predictor.java:177)
有没有大神用过DJL调用的, 怎么调用模型预测,怎么设置参数
0
收藏
请登录后评论