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

updates

This commit is contained in:
Glenn Jocher 2019-08-09 18:22:27 +02:00
parent 0c52cc0106
commit 298463e530
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train.py
View File

@ -193,6 +193,7 @@ def train(cfg,
('Epoch', 'gpu_mem', 'GIoU/xy', 'wh', 'obj', 'cls', 'total', 'targets', 'img_size')) ('Epoch', 'gpu_mem', 'GIoU/xy', 'wh', 'obj', 'cls', 'total', 'targets', 'img_size'))
# Update scheduler # Update scheduler
if epoch > 0:
scheduler.step() scheduler.step()
# Freeze backbone at epoch 0, unfreeze at epoch 1 (optional) # Freeze backbone at epoch 0, unfreeze at epoch 1 (optional)