File tree Expand file tree Collapse file tree 1 file changed +2
-5
lines changed
extension/training/optimizer Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Original file line number Diff line number Diff line change @@ -179,17 +179,14 @@ class ET_EXPERIMENTAL SGDParamGroup {
179
179
*/
180
180
class ET_EXPERIMENTAL SGD {
181
181
public:
182
- explicit SGD (
183
- const std::vector<SGDParamGroup>& param_groups,
184
- SGDOptions defaults)
182
+ SGD (const std::vector<SGDParamGroup>& param_groups, SGDOptions defaults)
185
183
: defaults_(std::make_unique<SGDOptions>(defaults)) {
186
184
for (const auto & param_group : param_groups) {
187
185
add_param_group (param_group);
188
186
}
189
187
}
190
188
191
- explicit SGD (
192
- const std::map<executorch::aten::string_view, executorch::aten::Tensor>&
189
+ SGD (const std::map<executorch::aten::string_view, executorch::aten::Tensor>&
193
190
named_parameters,
194
191
SGDOptions defaults)
195
192
: SGD({SGDParamGroup (named_parameters)}, defaults) {}
You can’t perform that action at this time.
0 commit comments