zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

create_backbone with SourceChangeWarning update

This commit is contained in:
Glenn Jocher 2020-06-07 15:45:05 -07:00
parent 546f973562
commit 1f698af6f1
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
utils/utils.py
View File

@ -557,7 +557,11 @@ def strip_optimizer(f='weights/best.pt'): # from utils.utils import *; strip_op
def create_backbone(f='weights/best.pt', s='weights/backbone.pt'): # from utils.utils import *; create_backbone() def create_backbone(f='weights/best.pt', s='weights/backbone.pt'): # from utils.utils import *; create_backbone()
# create backbone 's' from 'f' # create backbone 's' from 'f'
x = torch.load(f, map_location=torch.device('cpu')) device = torch.device('cpu')
x = torch.load(f, map_location=device)
torch.save(x, s) # update model if SourceChangeWarning
x = torch.load(s, map_location=device)
x['optimizer'] = None x['optimizer'] = None
x['training_results'] = None x['training_results'] = None
x['epoch'] = -1 x['epoch'] = -1