From fa7c517ece0719a03d0746db40a79ebd6c8ad3e1 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 14 Nov 2019 18:20:54 -0800 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 e4d67e4d..b3a15c5e 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -579,7 +579,7 @@ def print_model_biases(model): def strip_optimizer(f='weights/last.pt'): # from utils.utils import *; strip_optimizer() # Strip optimizer from *.pt files for lighter files (reduced by 2/3 size) x = torch.load(f, map_location=torch.device('cpu')) - # x['epoch'] = 0 # uncomment to create a backbone + # x['epoch'] = -1 # uncomment to create a backbone x['optimizer'] = None torch.save(x, f)