Fix minor bug on PolynomialLR invocation (#6436)

datumbox · web-flow · commit 78d680fe1424 · 2022-08-17T12:48:36.000+01:00
Resolves issue reported at 6e535db#commitcomment-81409388 There seems to be a misspelling on the name of the parameter. This PR updates `total_steps` to `total_iters` which is the correct argument.
diff --git a/references/segmentation/train.py b/references/segmentation/train.py
@@ -186,7 +186,7 @@ def main(args):
 
     iters_per_epoch = len(data_loader)
     main_lr_scheduler = PolynomialLR(
-        optimizer, total_steps=iters_per_epoch * (args.epochs - args.lr_warmup_epochs), power=0.9
+        optimizer, total_iters=iters_per_epoch * (args.epochs - args.lr_warmup_epochs), power=0.9
     )
 
     if args.lr_warmup_epochs > 0:

Original file line number	Diff line number	Diff line change
`@@ -186,7 +186,7 @@ def main(args):`
`186`	`186`
`187`	`187`	`iters_per_epoch = len(data_loader)`
`188`	`188`	`main_lr_scheduler = PolynomialLR(`
`189`		`- optimizer, total_steps=iters_per_epoch * (args.epochs - args.lr_warmup_epochs), power=0.9`
	`189`	`+ optimizer, total_iters=iters_per_epoch * (args.epochs - args.lr_warmup_epochs), power=0.9`
`190`	`190`	`)`
`191`	`191`
`192`	`192`	`if args.lr_warmup_epochs > 0:`