[fbsync] Fix minor bug on PolynomialLR invocation (#6436)

NicolasHug · facebook-github-bot · commit 4e602a8ce8a1 · 2022-08-24T04:05:22.000-07:00
Summary: 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. Reviewed By: datumbox Differential Revision: D38824253 fbshipit-source-id: 08401f694398a8c64e198bfc16fed874f5c7b462
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:`