diff --git a/extension/training/optimizer/sgd.h b/extension/training/optimizer/sgd.h index 055d561287d..b3e58ea8d8e 100644 --- a/extension/training/optimizer/sgd.h +++ b/extension/training/optimizer/sgd.h @@ -179,7 +179,7 @@ class ET_EXPERIMENTAL SGDParamGroup { */ class ET_EXPERIMENTAL SGD { public: - explicit SGD( + SGD( const std::vector& param_groups, SGDOptions defaults) : defaults_(std::make_unique(defaults)) { @@ -188,7 +188,7 @@ class ET_EXPERIMENTAL SGD { } } - explicit SGD( + SGD( const std::map& named_parameters, SGDOptions defaults)