From 8fd8d8eb04decd82084e5fbd5d5b7f570a1554c6 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 1 Sep 2018 18:41:05 +0200 Subject: [PATCH] updates --- README.md | 2 +- train.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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