added return statement (#2029)

1c7aa0c0 · PatrickBue · GitHub · 561a014b · 1c7aa0c0
Unverified Commit 1c7aa0c0 authored Mar 30, 2020 by PatrickBue Committed by GitHub Mar 30, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

references/detection/engine.py references/detection/engine.py +2 -0

No files found.
--- a/references/detection/engine.py
+++ b/references/detection/engine.py
@@ -52,6 +52,8 @@ def train_one_epoch(model, optimizer, data_loader, device, epoch, print_freq):
        metric_logger.update(loss=losses_reduced, **loss_dict_reduced)
        metric_logger.update(lr=optimizer.param_groups[0]["lr"])

+    return metric_logger
+

 def _get_iou_types(model):
    model_without_ddp = model