From a99cb674b6d3882b307b890f69d2fc8e7d220610 Mon Sep 17 00:00:00 2001 From: Joseph Date: Tue, 27 Oct 2020 04:42:52 +0530 Subject: [PATCH] Updates --- detectron2/modeling/roi_heads/fast_rcnn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/detectron2/modeling/roi_heads/fast_rcnn.py b/detectron2/modeling/roi_heads/fast_rcnn.py index 3d19e16..ea85511 100644 --- a/detectron2/modeling/roi_heads/fast_rcnn.py +++ b/detectron2/modeling/roi_heads/fast_rcnn.py @@ -667,7 +667,7 @@ class FastRCNNOutputLayers(nn.Module): new_means[index] = torch.tensor(item).mean(dim=0) # Update the MUs for i, mean in enumerate(self.means): - if(mean) is not None: + if(mean) is not None and new_means[i] is not None: self.means[i] = self.clustering_momentum * mean + \ (1 - self.clustering_momentum) * new_means[i]