如何在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
收藏
请登录后评论