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 6232ff30a747..1be6e3deef54 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.