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-10 00:54:55 +01:00
parent 966bc16d1a
commit e04bb75ff1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
models.py
View File

@ -139,8 +139,8 @@ class YOLOLayer(nn.Module):
# Training # Training
if targets is not None: if targets is not None:
MSELoss = nn.MSELoss(size_average=True) MSELoss = nn.MSELoss()
BCEWithLogitsLoss = nn.BCEWithLogitsLoss(size_average=True) BCEWithLogitsLoss = nn.BCEWithLogitsLoss()
CrossEntropyLoss = nn.CrossEntropyLoss() CrossEntropyLoss = nn.CrossEntropyLoss()
if requestPrecision: if requestPrecision: