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-05 16:19:51 +02:00
parent 65eccee4ef
commit 54e43b9ad6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -15,7 +15,7 @@ def train(
data_cfg, data_cfg,
img_size=416, img_size=416,
resume=False, resume=False,
epochs=270, epochs=273, # 500200 batches at bs 64, dataset length 117263
batch_size=16, batch_size=16,
accumulate=1, accumulate=1,
multi_scale=False, multi_scale=False,
@ -75,7 +75,7 @@ def train(
else: else:
cutoff = load_darknet_weights(model, weights + 'darknet53.conv.74') cutoff = load_darknet_weights(model, weights + 'darknet53.conv.74')
# Set scheduler (reduce lr at epoch 250) # Set scheduler (reduce lr at epochs 218, 245, i.e. batches 400k, 450k)
scheduler = torch.optim.lr_scheduler.MultiStepLR(optimizer, milestones=[218, 245], gamma=0.1, scheduler = torch.optim.lr_scheduler.MultiStepLR(optimizer, milestones=[218, 245], gamma=0.1,
last_epoch=start_epoch - 1) last_epoch=start_epoch - 1)