harness-race
/

control-r2

davanstrien HF Staff commited on Aug 7

Commit

7eacab2

verified ·

1 Parent(s): e91a8da

Upload train.py with huggingface_hub

Files changed (1) hide show

train.py CHANGED Viewed

@@ -121,7 +121,7 @@ class LRWarmupCosine:
 def evaluate(model, loader, processor, device, num_eval=None):
     model.eval()
-    metric = MeanAveragePrecision(class_metrics=True, iou_type="bbox")
     count = 0
     with torch.no_grad():
         for batch in tqdm(loader, desc="eval"):

 def evaluate(model, loader, processor, device, num_eval=None):
     model.eval()
+    metric = MeanAveragePrecision(class_metrics=True, iou_type="bbox", backend="faster_coco_eval")
     count = 0
     with torch.no_grad():
         for batch in tqdm(loader, desc="eval"):