From 6080fb60b6c7d9ed218cdaad39d57b8ba5e84c97 Mon Sep 17 00:00:00 2001 From: Frank Liu Date: Fri, 28 Jun 2024 10:37:49 -0700 Subject: [PATCH] [mxnet] Fixes GloveWordEmbeddingTranslator bug Fixes: #3286 --- .../zoo/nlp/embedding/GloveWordEmbeddingTranslatorFactory.java | 1 + 1 file changed, 1 insertion(+) diff --git a/engines/mxnet/mxnet-model-zoo/src/main/java/ai/djl/mxnet/zoo/nlp/embedding/GloveWordEmbeddingTranslatorFactory.java b/engines/mxnet/mxnet-model-zoo/src/main/java/ai/djl/mxnet/zoo/nlp/embedding/GloveWordEmbeddingTranslatorFactory.java index de9cb4ffc74..a5d6213c198 100644 --- a/engines/mxnet/mxnet-model-zoo/src/main/java/ai/djl/mxnet/zoo/nlp/embedding/GloveWordEmbeddingTranslatorFactory.java +++ b/engines/mxnet/mxnet-model-zoo/src/main/java/ai/djl/mxnet/zoo/nlp/embedding/GloveWordEmbeddingTranslatorFactory.java @@ -70,6 +70,7 @@ public void prepare(TranslatorContext ctx) { /** {@inheritDoc} */ @Override public NDList processOutput(TranslatorContext ctx, NDList list) { + list.detach(); return list; }