From ee4abc8cdf61a6f693bb3782e0bf359cdb143345 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 15 Feb 2019 14:07:55 +0200 Subject: [PATCH] optimize imports --- models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models.py b/models.py index a8d5e5db..d6a50eeb 100755 --- a/models.py +++ b/models.py @@ -316,7 +316,7 @@ def load_darknet_weights(self, weights, cutoff=-1): # Needed to write header when saving weights self.header_info = header - self.seen = header[3] + self.seen = header[3] # number of images seen during training weights = np.fromfile(fp, dtype=np.float32) # The rest are weights fp.close() @@ -365,7 +365,7 @@ def load_darknet_weights(self, weights, cutoff=-1): def save_weights(self, path, cutoff=-1): fp = open(path, 'wb') - self.header_info[3] = self.seen + self.header_info[3] = self.seen # number of images seen during training self.header_info.tofile(fp) # Iterate through layers