mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

update create_pretrained()

Browse Source
This commit is contained in:
Glenn Jocher 2020-07-10 00:25:35 -07:00
parent 1531795f5e
commit 41bb70b325
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/utils.py
View File

@ -641,9 +641,7 @@ def strip_optimizer(f='weights/best.pt'): # from utils.utils import *; strip_op
def create_pretrained(f='weights/best.pt', s='weights/pretrained.pt'): # from utils.utils import *; create_pretrained()
# create pretrained checkpoint 's' from 'f' (create_pretrained(x, x) for x in glob.glob('./*.pt'))
device = torch.device('cpu')
x = torch.load(s, map_location=device)
x = torch.load(f, map_location=torch.device('cpu'))
x['optimizer'] = None
x['training_results'] = None
x['epoch'] = -1