Skip to content

GitLab

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

Fix lint (#1693)

parent 013d1322
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
references/detection/train.py
View file @ d2c763e1
...@@ -115,7 +115,7 @@ def main(args): ...@@ -115,7 +115,7 @@ def main(args):
optimizer.load_state_dict(checkpoint['optimizer']) optimizer.load_state_dict(checkpoint['optimizer'])
lr_scheduler.load_state_dict(checkpoint['lr_scheduler']) lr_scheduler.load_state_dict(checkpoint['lr_scheduler'])
args.start_epoch = checkpoint['epoch'] + 1 args.start_epoch = checkpoint['epoch'] + 1
if args.test_only: if args.test_only:
evaluate(model, data_loader_test, device=device) evaluate(model, data_loader_test, device=device)
return return
......
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