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

updates

This commit is contained in:
Glenn Jocher 2019-04-27 18:36:19 +02:00
parent 1e3fb6566c
commit ccfd44c2f8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/utils.py
View File

@ -259,7 +259,7 @@ def compute_loss(p, targets, model): # predictions, targets, model
# Define criteria # Define criteria
MSE = nn.MSELoss() MSE = nn.MSELoss()
CE = nn.CrossEntropyLoss(weight=model.class_weights) CE = nn.CrossEntropyLoss() # (weight=model.class_weights)
BCE = nn.BCEWithLogitsLoss() BCE = nn.BCEWithLogitsLoss()
# Compute losses # Compute losses