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

EMA class updates

This commit is contained in:
Glenn Jocher 2020-03-14 17:33:29 -07:00
parent 5ebbb2db28
commit d91469a516
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -198,7 +198,7 @@ def train():
model.class_weights = labels_to_class_weights(dataset.labels, nc).to(device) # attach class weights model.class_weights = labels_to_class_weights(dataset.labels, nc).to(device) # attach class weights
# Model EMA # Model EMA
# ema = torch_utils.ModelEMA(model, decay=0.9997) # ema = torch_utils.ModelEMA(model, decay=0.9998)
# Start training # Start training
nb = len(dataloader) # number of batches nb = len(dataloader) # number of batches