From 5f1b78ad614b45c5a98e7afdd295e20033d5ad3c Mon Sep 17 00:00:00 2001
From: "Kin-Yiu, Wong" <102582011@cc.ncu.edu.tw>
Date: Fri, 8 Jul 2022 17:12:36 +0800
Subject: [PATCH] main code

set top_k temporally
---
 utils/loss.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/loss.py b/utils/loss.py
index 4e087826..31386328 100644
--- a/utils/loss.py
+++ b/utils/loss.py
@@ -712,7 +712,7 @@ class ComputeLossOTA:
 
             pair_wise_iou_loss = -torch.log(pair_wise_iou + 1e-8)
 
-            top_k, _ = torch.topk(pair_wise_iou, min(20, pair_wise_iou.shape[1]), dim=1)
+            top_k, _ = torch.topk(pair_wise_iou, min(10, pair_wise_iou.shape[1]), dim=1)
             dynamic_ks = torch.clamp(top_k.sum(1).int(), min=1)
 
             gt_cls_per_image = (