Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

--resume training from URL weights fix (#8884)

Browse Source 
--resume training from URL weights fix

@kalenmike should fix data error on HUB resume
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-08-05 16:42:10 +02:00
1 parent e073658e11
commit daed7a844e
1 file changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+2 -2
View File
@@ -496,8 +496,8 @@ def main(opt, callbacks=Callbacks()):
d = torch.load(last, map_location='cpu')['opt'] d = torch.load(last, map_location='cpu')['opt']
opt = argparse.Namespace(**d) # replace opt = argparse.Namespace(**d) # replace
opt.cfg, opt.weights, opt.resume = '', str(last), True # reinstate opt.cfg, opt.weights, opt.resume = '', str(last), True # reinstate
if is_url(opt.data): if is_url(opt_data):
opt.data = str(opt_data) # avoid HUB resume auth timeout opt.data = check_file(opt_data) # avoid HUB resume auth timeout
else: else:
opt.data, opt.cfg, opt.hyp, opt.weights, opt.project = \ opt.data, opt.cfg, opt.hyp, opt.weights, opt.project = \
check_file(opt.data), check_yaml(opt.cfg), check_yaml(opt.hyp), str(opt.weights), str(opt.project) # checks check_file(opt.data), check_yaml(opt.cfg), check_yaml(opt.hyp), str(opt.weights), str(opt.project) # checks