update error message in AllReduceParameter (#2997)

* update error message in AllReduceParameter
dding3 · Apr 9, 2020 · 7b29633 · 7b29633
1 parent 148465f
commit 7b29633
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.../src/main/scala/com/intel/analytics/bigdl/dllib/optim/parameters/AllReduceParameter.scala b/.../src/main/scala/com/intel/analytics/bigdl/dllib/optim/parameters/AllReduceParameter.scala
@@ -207,7 +207,8 @@ class AllReduceParameter[T: ClassTag](
               val blockId = getWeightBlockId(pid)
               val localBuffer = BlockManagerWrapper.getLocalOrRemoteBytes(blockId).getOrElse {
                 throw new RuntimeException(s"Didn't find weight block $blockId in the block " +
-                  s"manager. Did you initialize this AllReduceParameter on every executor?")
+                  s"manager. This is usually because executor crashed. Please check your" +
+                  s"executors' log see the error (usually an OutOfMemory error)")
               }
               val start = pid * taskSize + math.min(pid, extraSize)
               val length = taskSize + (if (pid < extraSize) 1 else 0)