diff --git a/scala/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/models/inception/Train.scala b/scala/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/models/inception/Train.scala index 3d9cecc6d4f..38ce1212f1d 100644 --- a/scala/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/models/inception/Train.scala +++ b/scala/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/models/inception/Train.scala @@ -77,7 +77,7 @@ object TrainInceptionV1 { case "optimizerv2" => Engine.setOptimizerVersion(OptimizerV2) } } - + val optimMethod = if (param.stateSnapshot.isDefined) { OptimMethod.load[Float](param.stateSnapshot.get) } else {