diff --git a/README.md b/README.md index 8af276ec..4dc3fa8e 100755 --- a/README.md +++ b/README.md @@ -19,7 +19,7 @@ Python 3.6 or later with the following `pip3 install -U -r requirements.txt` pac # Training -Run `train.py` to begin training after downloading COCO data with `data/get_coco_dataset.sh`. +Run `train.py` to begin training after downloading COCO data with `data/get_coco_dataset.sh` and specifying COCO path on line 37 (local) or line 39 (cloud). Run `train.py -resume 1` to resume training from the most recently saved checkpoint `checkpoints/latest.pt`. diff --git a/train.py b/train.py index 4a8cc70b..c70b5af7 100644 --- a/train.py +++ b/train.py @@ -33,9 +33,9 @@ def main(opt): # Configure run data_config = parse_data_config(opt.data_config_path) num_classes = int(data_config['classes']) - if platform == 'darwin': # macos + if platform == 'darwin': # MacOS (local) train_path = data_config['valid'] - else: # linux (gcp cloud) + else: # linux (cloud, i.e. gcp) train_path = '../coco/trainvalno5k.part' # Initialize model