From 4843cc4e087d7e7a09371e0144eae8902a26ee8d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 27 Dec 2019 12:52:01 -0800 Subject: [PATCH] updates --- train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/train.py b/train.py index 5fb31e25..3e903c03 100644 --- a/train.py +++ b/train.py @@ -192,7 +192,7 @@ def train(): augment=True, hyp=hyp, # augmentation hyperparameters rect=opt.rect, # rectangular training - image_weights=opt.img_weights, + image_weights=False, cache_labels=epochs > 10, cache_images=opt.cache_images and not opt.prebias) @@ -430,7 +430,6 @@ if __name__ == '__main__': parser.add_argument('--notest', action='store_true', help='only test final epoch') parser.add_argument('--evolve', action='store_true', help='evolve hyperparameters') parser.add_argument('--bucket', type=str, default='', help='gsutil bucket') - parser.add_argument('--img-weights', action='store_true', help='select training images by weight') parser.add_argument('--cache-images', action='store_true', help='cache images for faster training') parser.add_argument('--weights', type=str, default='weights/ultralytics68.pt', help='initial weights') parser.add_argument('--arc', type=str, default='default', help='yolo architecture') # defaultpw, uCE, uBCE