From b52a49cf129ca3790d80e03bf53b1ceb5f8d123a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 15 Dec 2018 21:06:39 +0100 Subject: [PATCH] updates --- models.py | 1 + 1 file changed, 1 insertion(+) diff --git a/models.py b/models.py index 1d105709..c754ab8f 100755 --- a/models.py +++ b/models.py @@ -232,6 +232,7 @@ class Darknet(nn.Module): def __init__(self, cfg_path, img_size=416): super(Darknet, self).__init__() + self.module_defs = parse_model_config(cfg_path) self.module_defs[0]['height'] = img_size self.hyperparams, self.module_list = create_modules(self.module_defs)