Adding blanks to make the codes look better

zhiqwang · Aug 22, 2021 · c4b9bf2 · c4b9bf2
1 parent b2a14f1
commit c4b9bf2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tools/eval_metric.py b/tools/eval_metric.py
@@ -129,13 +129,15 @@ def evaluate(model, data_loader, coco_evaluator, device, print_freq):
 
         if torch.cuda.is_available():
             torch.cuda.synchronize()
-        model_time = time.time()
 
+        model_time = time.time()
         preds = model(images)
         model_time = time.time() - model_time
+
         evaluator_time = time.time()
         coco_evaluator.update(preds, targets)
         evaluator_time = time.time() - evaluator_time
+
         metric_logger.update(model_time=model_time, evaluator_time=evaluator_time)
 
     # gather the stats from all processes