From fe9678093e08fcf339ef848b9274a90a7c29495c Mon Sep 17 00:00:00 2001 From: Le-Zheng <30695225+Le-Zheng@users.noreply.github.com> Date: Tue, 22 Jun 2021 02:49:35 +0100 Subject: [PATCH] make default DistriOptimizer as V2 (#3129) * make default DistriOptimizer as V2 * update --- .../main/scala/com/intel/analytics/bigdl/common/Engine.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dl/src/main/scala/com/intel/analytics/bigdl/common/Engine.scala b/dl/src/main/scala/com/intel/analytics/bigdl/common/Engine.scala index 2b063160010..ff03e2b3678 100644 --- a/dl/src/main/scala/com/intel/analytics/bigdl/common/Engine.scala +++ b/dl/src/main/scala/com/intel/analytics/bigdl/common/Engine.scala @@ -225,10 +225,10 @@ object Engine { /** * Notice: Please use property bigdl.optimizerVersion to set optimizerVersion. - * Default version is OptimizerV1 + * Default version is OptimizerV2 */ private var optimizerVersion: OptimizerVersion = { - System.getProperty("bigdl.optimizerVersion", "optimizerv1").toLowerCase(Locale.ROOT) match { + System.getProperty("bigdl.optimizerVersion", "optimizerv2").toLowerCase(Locale.ROOT) match { case "optimizerv1" => OptimizerV1 case "optimizerv2" => OptimizerV2 case optimizerVersion => throw new IllegalArgumentException(s"Unknown type $optimizerVersion")