From 41bb70b32598c016cc55debae7bafd26ec94e984 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 10 Jul 2020 00:25:35 -0700 Subject: [PATCH] update create_pretrained() --- utils/utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/utils/utils.py b/utils/utils.py index 53298a4..2069749 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -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