Skip to content

Commit

Permalink
Dont return lr in load_optim
Browse files Browse the repository at this point in the history
  • Loading branch information
adityac8 authored Mar 22, 2021
1 parent ca72762 commit 5db6dbb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions utils/model_utils.py
Original file line number Diff line number Diff line change
@@ -49,5 +49,5 @@ def load_start_epoch(weights):
def load_optim(optimizer, weights):
checkpoint = torch.load(weights)
optimizer.load_state_dict(checkpoint['optimizer'])
for p in optimizer.param_groups: lr = p['lr']
return lr
# for p in optimizer.param_groups: lr = p['lr']
# return lr

0 comments on commit 5db6dbb

Please sign in to comment.