首页 Paddle框架 帖子详情
如何在DJL中调用paddle语言处理模型
收藏
快速回复
Paddle框架 问答深度学习 376 0
如何在DJL中调用paddle语言处理模型
收藏
快速回复
Paddle框架 问答深度学习 376 0

【问题描述】
想把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
收藏
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户