From 1f943e886f1bd0a95764c30b941d812291824a63 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 7 Dec 2019 15:17:29 -0800 Subject: [PATCH] updates --- train.py | 1 - 1 file changed, 1 deletion(-) diff --git a/train.py b/train.py index e22789ee..ef45c5dd 100644 --- a/train.py +++ b/train.py @@ -114,7 +114,6 @@ def train(): try: chkpt['model'] = {k: v for k, v in chkpt['model'].items() if model.state_dict()[k].numel() == v.numel()} model.load_state_dict(chkpt['model'], strict=False) - # model.load_state_dict(chkpt['model']) except KeyError as e: s = "%s is not compatible with %s. Specify --weights '' or specify a --cfg compatible with %s. " \ "See https://github.com/ultralytics/yolov3/issues/657" % (opt.weights, opt.cfg, opt.weights)