Skip to content

GitLab

Unverified Commit 013d1322 authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Fix lint (#1692)

parent f4a82243
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
references/detection/train.py
View file @ 013d1322
...@@ -108,7 +108,7 @@ def main(args): ...@@ -108,7 +108,7 @@ def main(args):
# lr_scheduler = torch.optim.lr_scheduler.StepLR(optimizer, step_size=args.lr_step_size, gamma=args.lr_gamma) # lr_scheduler = torch.optim.lr_scheduler.StepLR(optimizer, step_size=args.lr_step_size, gamma=args.lr_gamma)
lr_scheduler = torch.optim.lr_scheduler.MultiStepLR(optimizer, milestones=args.lr_steps, gamma=args.lr_gamma) lr_scheduler = torch.optim.lr_scheduler.MultiStepLR(optimizer, milestones=args.lr_steps, gamma=args.lr_gamma)
if args.resume: if args.resume:
checkpoint = torch.load(args.resume, map_location='cpu') checkpoint = torch.load(args.resume, map_location='cpu')
model_without_ddp.load_state_dict(checkpoint['model']) model_without_ddp.load_state_dict(checkpoint['model'])
......
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