From b5bfc30759414104c3e301590f463318740aafe8 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 18 Apr 2019 22:40:32 +0200 Subject: [PATCH] updates --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index dd1b2fcf..abe0400f 100755 --- a/README.md +++ b/README.md @@ -43,9 +43,9 @@ Python 3.7 or later with the following `pip3 install -U -r requirements.txt` pac # Training -**Start Training:** Run `train.py` to begin training after downloading COCO data with `data/get_coco_dataset.sh`. +**Start Training:** `python3 train.py` to begin training after downloading COCO data with `data/get_coco_dataset.sh`. -**Resume Training:** Run `train.py --resume` resumes training from the latest checkpoint `weights/latest.pt`. +**Resume Training:** `python3 train.py --resume` to resume training from `weights/latest.pt`. Each epoch trains on 117,263 images from the train and validate COCO sets, and tests on 5000 images from the COCO validate set. Default training settings produce loss plots below, with **training speed of 0.25 s/batch on a V100 GPU (almost 50 COCO epochs/day)**.