glenn-jocher commited on
Commit
0362f10
1 Parent(s): 94e6711
Files changed (1) hide show
  1. utils/utils.py +1 -1
utils/utils.py CHANGED
@@ -547,7 +547,7 @@ def non_max_suppression(prediction, conf_thres=0.1, iou_thres=0.6, fast=False, c
547
 
548
 
549
  def strip_optimizer(f='weights/best.pt'): # from utils.utils import *; strip_optimizer()
550
- # Strip optimizer from *.pt files for lighter files (reduced by 2/3 size)
551
  x = torch.load(f, map_location=torch.device('cpu'))
552
  x['optimizer'] = None
553
  torch.save(x, f)
 
547
 
548
 
549
  def strip_optimizer(f='weights/best.pt'): # from utils.utils import *; strip_optimizer()
550
+ # Strip optimizer from *.pt files for lighter files (reduced by 1/2 size)
551
  x = torch.load(f, map_location=torch.device('cpu'))
552
  x['optimizer'] = None
553
  torch.save(x, f)