updates
This commit is contained in:
parent
d35cff2d22
commit
defae83d77
2
train.py
2
train.py
|
@ -34,7 +34,7 @@ def main(opt):
|
|||
data_config = parse_data_config(opt.data_config_path)
|
||||
num_classes = int(data_config['classes'])
|
||||
if platform == 'darwin': # MacOS (local)
|
||||
train_path = data_config['valid']
|
||||
train_path = data_config['train']
|
||||
else: # linux (cloud, i.e. gcp)
|
||||
train_path = '../coco/trainvalno5k.part'
|
||||
|
||||
|
|
Loading…
Reference in New Issue