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

updates

This commit is contained in:
Glenn Jocher 2019-03-21 22:49:57 +02:00
parent cd81f978be
commit 176851f83a
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

@ -43,7 +43,7 @@ def train(
# Dataloader
dataset = LoadImagesAndLabels(train_path, img_size=img_size, augment=True)
dataloader = DataLoader(dataset, batch_size=batch_size, num_workers=4)
dataloader = DataLoader(dataset, batch_size=batch_size, num_workers=num_workers)
cutoff = -1 # backbone reaches to cutoff layer
start_epoch = 0