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 17:52:43 +02:00
parent d25190e15b
commit 469dede6d1
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

@ -66,7 +66,7 @@ def train(
# Configure run # Configure run
data_cfg = parse_data_cfg(data_cfg) data_cfg = parse_data_cfg(data_cfg)
train_path = data_cfg['train'] train_path = data_cfg['train']
nc = data_cfg['classes'] # number of classes nc = int(data_cfg['classes']) # number of classes
# Initialize model # Initialize model
model = Darknet(cfg, img_size).to(device) model = Darknet(cfg, img_size).to(device)