"projects/vscode:/vscode.git/clone" did not exist on "46e82efb4e865e24a1351f20a16cf74695cfed5d"
Unverified Commit 78d680fe authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Fix minor bug on `PolynomialLR` invocation (#6436)

Resolves issue reported at https://github.com/pytorch/vision/commit/6e535db255cee3ce878dd7a54dda01d4ec8932c1#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.
parent db718021
......@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment