Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

update create_pretrained()

Browse Source
This commit is contained in:
Glenn Jocher committed 2020-07-10 00:25:35 -07:00
1 parent 1531795f5e
commit 41bb70b325
1 file changed
+1 -3
Download Patch File Download Diff File Expand all files Collapse all files
utils/utils.py
+1 -3
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() 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')) # create pretrained checkpoint 's' from 'f' (create_pretrained(x, x) for x in glob.glob('./*.pt'))
device = torch.device('cpu') x = torch.load(f, map_location=torch.device('cpu'))
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