Fixes Scala memory leak (apache#10436)

andrewfayres · Jun 9, 2018 · 29e696a · 29e696a
1 parent b32e04b
commit 29e696a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scala-package/core/src/main/scala/org/apache/mxnet/FeedForward.scala b/scala-package/core/src/main/scala/org/apache/mxnet/FeedForward.scala
@@ -230,8 +230,9 @@ class FeedForward private(
       val padded = batch.pad
       val realSize = batchSize - padded
       for ((list, nd) <- outputs zip predExec.outputs) {
-        list += nd.slice(0, realSize).copy()
+        list += nd.slice(0, realSize)
       }
+      batch.dispose()
     }
     // TODO(Yizhi): we can use Symbol.concat to do the same thing. Can it be more efficient?
     val results = outputs.map(NDArray.concatenate(_))