apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updates

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
models.py
+1 -1
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)