Skip to content

Commit

Permalink
Is it repeated here?
Browse files Browse the repository at this point in the history
  • Loading branch information
CottonCandyZ authored Apr 12, 2023
1 parent 9fff385 commit 0a2ac0f
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions processor/processor.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,8 +50,6 @@ def do_train(start_epoch, args, model, train_loader, evaluator, optimizer,
ret = model(batch)
total_loss = sum([v for k, v in ret.items() if "loss" in k])

total_loss = sum([v for k, v in ret.items() if "loss" in k])

batch_size = batch['images'].shape[0]
meters['loss'].update(total_loss.item(), batch_size)
meters['sdm_loss'].update(ret.get('sdm_loss', 0), batch_size)
Expand Down

0 comments on commit 0a2ac0f

Please sign in to comment.