added return statement (#2029)

pytorch · Mar 30, 2020 · 1c7aa0c · 1c7aa0c
1 parent 561a014
commit 1c7aa0c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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