mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update train.py incompatible model message fix #222

pull/244/head
Glenn Jocher 2020-06-29 12:45:25 -07:00
parent 01895f3924
commit b203c9b7ff
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
train.py
View File

@ -118,8 +118,9 @@ def train(hyp):
if model.state_dict()[k].shape == v.shape} # to FP32, filter if model.state_dict()[k].shape == v.shape} # to FP32, filter
model.load_state_dict(ckpt['model'], strict=False) model.load_state_dict(ckpt['model'], strict=False)
except KeyError as e: except KeyError as e:
s = "%s is not compatible with %s. Specify --weights '' or specify a --cfg compatible with %s." \ s = "%s is not compatible with %s. This may be due to model differences or %s may be out of date. " \
% (opt.weights, opt.cfg, opt.weights) "Please delete or update %s and try again, or use --weights '' to train from scatch." \
% (opt.weights, opt.cfg, opt.weights, opt.weights)
raise KeyError(s) from e raise KeyError(s) from e
# load optimizer # load optimizer