diff --git a/train.py b/train.py index 6fff74c5..ca6090c5 100644 --- a/train.py +++ b/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'