This commit is contained in:
Glenn Jocher 2018-09-01 18:41:05 +02:00
parent 382100307a
commit 8fd8d8eb04
2 changed files with 3 additions and 3 deletions

View File

@ -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`.

View File

@ -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