Remove unused variable

2021-06-01 23:39:39 +02:00 · 2021-06-01 23:39:39 +02:00 · e8cd4d765c
commit e8cd4d765c
parent 8403b01081
1 changed files with 0 additions and 1 deletions
--- 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)