Uploaded image for project: 'Spark'
  1. Spark
  2. SPARK-16440

Undeleted broadcast variables in Word2Vec causing OoM for long runs

    XMLWordPrintableJSON

    Details

    • Type: Bug
    • Status: Resolved
    • Priority: Major
    • Resolution: Fixed
    • Affects Version/s: 1.6.0, 1.6.1, 1.6.2, 2.0.0
    • Fix Version/s: 1.6.3, 2.0.1
    • Component/s: MLlib
    • Labels:
      None

      Description

      Three broadcast variables created at the beginning of Word2Vec.fit() are never deleted nor unpersisted. This seems to cause excessive memory consumption on the driver for a job running hundreds of successive training.

      They are

          val expTable = sc.broadcast(createExpTable())
          val bcVocab = sc.broadcast(vocab)
          val bcVocabHash = sc.broadcast(vocabHash)
      

        Attachments

          Issue Links

            Activity

              People

              • Assignee:
                anthony-truchet Anthony Truchet
                Reporter:
                anthony-truchet Anthony Truchet
              • Votes:
                0 Vote for this issue
                Watchers:
                2 Start watching this issue

                Dates

                • Created:
                  Updated:
                  Resolved:

                  Time Tracking

                  Estimated:
                  Original Estimate - 4h
                  4h
                  Remaining:
                  Remaining Estimate - 4h
                  4h
                  Logged:
                  Time Spent - Not Specified
                  Not Specified