glenn-jocher
commited on
Commit
•
0362f10
1
Parent(s):
94e6711
updates
Browse files- 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
|
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)
|