From 10145833c3a86add544f0291deaf860b61d43ea8 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 18 Jun 2020 00:16:47 -0700 Subject: [PATCH] strip_optimizer() default to save FP16 model --- utils/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/utils/utils.py b/utils/utils.py index 9c3d7d2..6bde407 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -620,6 +620,7 @@ def strip_optimizer(f='weights/best.pt'): # from utils.utils import *; strip_op # 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 + x['model'].half() # to FP16 torch.save(x, f) print('Optimizer stripped from %s' % f)