--resume training from URL weights fix (#8884)
--resume training from URL weights fix @kalenmike should fix data error on HUB resume
This commit is contained in:
1 parent
e073658e11
commit
daed7a844e
1 file changed
+2
-2
@@ -496,8 +496,8 @@ def main(opt, callbacks=Callbacks()):
|
||||
d = torch.load(last, map_location='cpu')['opt']
|
||||
opt = argparse.Namespace(**d) # replace
|
||||
opt.cfg, opt.weights, opt.resume = '', str(last), True # reinstate
|
||||
if is_url(opt.data):
|
||||
opt.data = str(opt_data) # avoid HUB resume auth timeout
|
||||
if is_url(opt_data):
|
||||
opt.data = check_file(opt_data) # avoid HUB resume auth timeout
|
||||
else:
|
||||
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
|
||||
|
||||
Reference in new issue
Block a user