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

updates

This commit is contained in:
Glenn Jocher 2018-09-03 00:35:46 +02:00
parent d35cff2d22
commit defae83d77
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

@ -34,7 +34,7 @@ def main(opt):
data_config = parse_data_config(opt.data_config_path) data_config = parse_data_config(opt.data_config_path)
num_classes = int(data_config['classes']) num_classes = int(data_config['classes'])
if platform == 'darwin': # MacOS (local) if platform == 'darwin': # MacOS (local)
train_path = data_config['valid'] train_path = data_config['train']
else: # linux (cloud, i.e. gcp) else: # linux (cloud, i.e. gcp)
train_path = '../coco/trainvalno5k.part' train_path = '../coco/trainvalno5k.part'