From e8cd4d765c6d24bc36694fc125883248227c58d5 Mon Sep 17 00:00:00 2001 From: Jensun Ravichandran Date: Tue, 1 Jun 2021 23:39:39 +0200 Subject: [PATCH] Remove unused variable --- prototorch/models/cbc.py | 1 - 1 file changed, 1 deletion(-) diff --git a/prototorch/models/cbc.py b/prototorch/models/cbc.py index d8d1800..59e7251 100644 --- a/prototorch/models/cbc.py +++ b/prototorch/models/cbc.py @@ -78,7 +78,6 @@ class ReasoningLayer(torch.nn.Module): def forward(self, detections): pk = self.reasoning_probabilities[0].clamp(0, 1) nk = (1 - pk) * self.reasoning_probabilities[1].clamp(0, 1) - epsilon = torch.finfo(pk.dtype).eps numerator = (detections @ (pk - nk)) + nk.sum(1) probs = numerator / (pk + nk).sum(1) probs = probs.squeeze(0)