From a444d1e3985611ab93629c5a1aff35da89deb057 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 --- Engine.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Engine.scala b/Engine.scala index 2b063160010..ff03e2b3678 100644 --- a/Engine.scala +++ b/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")