updates

2018-12-15 21:06:39 +01:00 · 2018-12-15 21:06:39 +01:00 · b52a49cf12
parent b079c1b10c
commit b52a49cf12
1 changed files with 1 additions and 0 deletions
--- 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)