Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2018-11-22 14:48:57 +01:00
parent bec94be01a
commit 959c67b4ed
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -273,9 +273,9 @@ class Darknet(nn.Module):
self.losses['TP'] = metrics[0].sum() self.losses['TP'] = metrics[0].sum()
self.losses['FP'] = metrics[1].sum() self.losses['FP'] = metrics[1].sum()
self.losses['FN'] = metrics[2].sum() self.losses['FN'] = metrics[2].sum()
self.losses['TC'] = 0
self.losses['metrics'] = metrics self.losses['metrics'] = metrics
self.losses['TC'] = 0
return sum(output) if is_training else torch.cat(output, 1) return sum(output) if is_training else torch.cat(output, 1)