From 9daa5e858a7fc9d2b07aae3e59d630feb461b390 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 14 Nov 2019 17:22:09 -0800 Subject: [PATCH] updates --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 3350f3a8..5f61f7c4 100644 --- a/train.py +++ b/train.py @@ -103,7 +103,7 @@ def train(): best_fitness = float('inf') attempt_download(weights) if weights.endswith('.pt'): # pytorch format - # possible weights are 'last.pt', 'yolov3-spp.pt', 'yolov3-tiny.pt' etc. + # possible weights are '*.pt', 'yolov3-spp.pt', 'yolov3-tiny.pt' etc. if opt.bucket: os.system('gsutil cp gs://%s/last.pt %s' % (opt.bucket, last)) # download from bucket chkpt = torch.load(weights, map_location=device) @@ -129,7 +129,7 @@ def train(): del chkpt elif len(weights) > 0: # darknet format - # possible weights are 'yolov3.weights', 'yolov3-tiny.conv.15', 'darknet53.conv.74' etc. + # possible weights are '*.weights', 'yolov3-tiny.conv.15', 'darknet53.conv.74' etc. cutoff = load_darknet_weights(model, weights) if opt.transfer or opt.prebias: # transfer learning edge (yolo) layers