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

Fix lint (#1693)

parent 013d1322
......@@ -115,7 +115,7 @@ def main(args):
optimizer.load_state_dict(checkpoint['optimizer'])
lr_scheduler.load_state_dict(checkpoint['lr_scheduler'])
args.start_epoch = checkpoint['epoch'] + 1
if args.test_only:
evaluate(model, data_loader_test, device=device)
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