We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f4a8224 commit 013d132Copy full SHA for 013d132
references/detection/train.py
@@ -108,7 +108,7 @@ def main(args):
108
109
# lr_scheduler = torch.optim.lr_scheduler.StepLR(optimizer, step_size=args.lr_step_size, gamma=args.lr_gamma)
110
lr_scheduler = torch.optim.lr_scheduler.MultiStepLR(optimizer, milestones=args.lr_steps, gamma=args.lr_gamma)
111
-
+
112
if args.resume:
113
checkpoint = torch.load(args.resume, map_location='cpu')
114
model_without_ddp.load_state_dict(checkpoint['model'])
0 commit comments