julius/prototorch_models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove unused variable

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-06-01 23:39:39 +02:00
parent 8403b01081
commit e8cd4d765c
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
prototorch/models/cbc.py
View File

@ -78,7 +78,6 @@ class ReasoningLayer(torch.nn.Module):
def forward(self, detections): def forward(self, detections):
pk = self.reasoning_probabilities[0].clamp(0, 1) pk = self.reasoning_probabilities[0].clamp(0, 1)
nk = (1 - pk) * self.reasoning_probabilities[1].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) numerator = (detections @ (pk - nk)) + nk.sum(1)
probs = numerator / (pk + nk).sum(1) probs = numerator / (pk + nk).sum(1)
probs = probs.squeeze(0) probs = probs.squeeze(0)