refactor: get_optimizer_parameters

kozistr · kozistr · commit db70700a9c11 · 2021-10-09T20:50:33.000+09:00
diff --git a/pytorch_optimizer/utils.py b/pytorch_optimizer/utils.py
@@ -49,12 +49,10 @@ def get_optimizer_parameters(
 ) -> PARAMETERS:
     param_optimizer: List[Tuple[str, nn.Parameter]] = list(model.named_parameters())
 
-    optimizer_parameters = [
+    return [
         {
             'params': [p for n, p in param_optimizer if not any(nd in n for nd in wd_ban_list)],
             'weight_decay': weight_decay,
         },
         {'params': [p for n, p in param_optimizer if any(nd in n for nd in wd_ban_list)], 'weight_decay': 0.0},
     ]
-
-    return optimizer_parameters

Original file line number	Diff line number	Diff line change
`@@ -49,12 +49,10 @@ def get_optimizer_parameters(`
`49`	`49`	`) -> PARAMETERS:`
`50`	`50`	`param_optimizer: List[Tuple[str, nn.Parameter]] = list(model.named_parameters())`
`51`	`51`
`52`		`- optimizer_parameters = [`
	`52`	`+ return [`
`53`	`53`	`{`
`54`	`54`	`'params': [p for n, p in param_optimizer if not any(nd in n for nd in wd_ban_list)],`
`55`	`55`	`'weight_decay': weight_decay,`
`56`	`56`	`},`
`57`	`57`	`{'params': [p for n, p in param_optimizer if any(nd in n for nd in wd_ban_list)], 'weight_decay': 0.0},`
`58`	`58`	`]`
`59`		`-`
`60`		`- return optimizer_parameters`