提交 41bb70b3 authored 作者: Glenn Jocher's avatar Glenn Jocher

update create_pretrained()

上级 1531795f
...@@ -641,9 +641,7 @@ def strip_optimizer(f='weights/best.pt'): # from utils.utils import *; strip_op ...@@ -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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论