From 7aa7976c695ba9519164c2e5f76ff2faed373960 Mon Sep 17 00:00:00 2001 From: Anthony Truchet Date: Mon, 28 Nov 2016 17:22:23 +0100 Subject: [PATCH] [SPARK-18612][MLLIB] Delete broadcasted variable in LBFGS CostFun This contribution is made on behalf of Criteo S.A. (http://labs.criteo.com/) under the terms of the Apache v2 License. --- .../main/scala/org/apache/spark/mllib/optimization/LBFGS.scala | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mllib/src/main/scala/org/apache/spark/mllib/optimization/LBFGS.scala b/mllib/src/main/scala/org/apache/spark/mllib/optimization/LBFGS.scala index 6232ff30a747e..1be6e3deef547 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/optimization/LBFGS.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/optimization/LBFGS.scala @@ -253,6 +253,9 @@ object LBFGS extends Logging { (grad1, loss1 + loss2) }) + // broadcasted model is not needed anymore + bcW.destroy() + /** * regVal is sum of weight squares if it's L2 updater; * for other updater, the same logic is followed.