print('Optimizer stripped from %s' % f)

2020-05-24 20:30:30 -07:00 · 2020-05-24 20:30:30 -07:00 · d6d6fb5e5b
parent 23f85a68b8
commit d6d6fb5e5b
1 changed files with 1 additions and 0 deletions
--- a/utils/utils.py
+++ b/utils/utils.py
@ -570,6 +570,7 @@ def strip_optimizer(f='weights/best.pt'):  # from utils.utils import *; strip_op
    # Strip optimizer from *.pt files for lighter files (reduced by 2/3 size)
    x = torch.load(f, map_location=torch.device('cpu'))
    x['optimizer'] = None
+    print('Optimizer stripped from %s' % f)
    torch.save(x, f)