diff --git a/extension/training/optimizer/sgd.h b/extension/training/optimizer/sgd.h index 055d561287d..9e3ab42817a 100644 --- a/extension/training/optimizer/sgd.h +++ b/extension/training/optimizer/sgd.h @@ -179,17 +179,14 @@ class ET_EXPERIMENTAL SGDParamGroup { */ class ET_EXPERIMENTAL SGD { public: - explicit SGD( - const std::vector& param_groups, - SGDOptions defaults) + SGD(const std::vector& param_groups, SGDOptions defaults) : defaults_(std::make_unique(defaults)) { for (const auto& param_group : param_groups) { add_param_group(param_group); } } - explicit SGD( - const std::map& + SGD(const std::map& named_parameters, SGDOptions defaults) : SGD({SGDParamGroup(named_parameters)}, defaults) {}