mirror of https://github.com/WongKinYiu/yolov7.git
Merge e89974e4c9
into a207844b1c
commit
b4f20a583b
|
@ -249,7 +249,7 @@ def attempt_load(weights, map_location=None):
|
|||
model = Ensemble()
|
||||
for w in weights if isinstance(weights, list) else [weights]:
|
||||
attempt_download(w)
|
||||
ckpt = torch.load(w, map_location=map_location) # load
|
||||
ckpt = torch.load(w, map_location=map_location, weights_only = False) # load
|
||||
model.append(ckpt['ema' if ckpt.get('ema') else 'model'].float().fuse().eval()) # FP32 model
|
||||
|
||||
# Compatibility updates
|
||||
|
|
Loading…
Reference in New Issue