From cc0411c0dc17e3239bac15f420d473d217c9e4e8 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 9 Sep 2021 18:01:59 +0200 Subject: [PATCH] Fix `--data from_HUB.zip` (#4732) @KalenMike --- train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/train.py b/train.py index c32664832d8b..e5410eeeba9f 100644 --- a/train.py +++ b/train.py @@ -36,7 +36,7 @@ from utils.datasets import create_dataloader from utils.general import labels_to_class_weights, increment_path, labels_to_image_weights, init_seeds, \ strip_optimizer, get_latest_run, check_dataset, check_git_status, check_img_size, check_requirements, \ - check_yaml, check_suffix, print_mutation, set_logging, one_cycle, colorstr, methods + check_file, check_yaml, check_suffix, print_mutation, set_logging, one_cycle, colorstr, methods from utils.downloads import attempt_download from utils.loss import ComputeLoss from utils.plots import plot_labels, plot_evolve @@ -105,6 +105,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary is_coco = data.endswith('coco.yaml') and nc == 80 # COCO dataset # Model + check_suffix(weights, '.pt') # check weights pretrained = weights.endswith('.pt') if pretrained: with torch_distributed_zero_first(RANK): @@ -484,8 +485,7 @@ def main(opt, callbacks=Callbacks()): opt.cfg, opt.weights, opt.resume = '', ckpt, True # reinstate LOGGER.info(f'Resuming training from {ckpt}') else: - check_suffix(opt.weights, '.pt') # check weights - opt.data, opt.cfg, opt.hyp = check_yaml(opt.data), check_yaml(opt.cfg), check_yaml(opt.hyp) # check YAMLs + opt.data, opt.cfg, opt.hyp = check_file(opt.data), check_yaml(opt.cfg), check_yaml(opt.hyp) # check YAMLs assert len(opt.cfg) or len(opt.weights), 'either --cfg or --weights must be specified' if opt.evolve: opt.project = 'runs/evolve'