From f6775ae1bccf860c86ae305da6dde4e0257ae17d Mon Sep 17 00:00:00 2001
From: PatrickBue <pabuehle@microsoft.com>
Date: Mon, 30 Mar 2020 08:58:18 -0400
Subject: [PATCH] added return statement

---
 references/detection/engine.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/references/detection/engine.py b/references/detection/engine.py
index 68c39a4fc1b..86c25d9c486 100644
--- 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