From 0362f103efede32fbc7a462d38bedb53ea81d46e Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 4 Jun 2020 10:19:06 -0700 Subject: [PATCH] updates --- utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/utils.py b/utils/utils.py index 3899231..e22cf14 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -547,7 +547,7 @@ def non_max_suppression(prediction, conf_thres=0.1, iou_thres=0.6, fast=False, c def strip_optimizer(f='weights/best.pt'): # from utils.utils import *; strip_optimizer() - # Strip optimizer from *.pt files for lighter files (reduced by 2/3 size) + # Strip optimizer from *.pt files for lighter files (reduced by 1/2 size) x = torch.load(f, map_location=torch.device('cpu')) x['optimizer'] = None torch.save(x, f)