hank1996 commited on
Commit
86a0c0e
·
1 Parent(s): f1c5cc4

Update models/experimental.py

Browse files
Files changed (1) hide show
  1. models/experimental.py +1 -1
models/experimental.py CHANGED
@@ -85,7 +85,7 @@ def attempt_load(weights, map_location=None):
85
  # Loads an ensemble of models weights=[a,b,c] or a single model weights=[a] or weights=a
86
  model = Ensemble()
87
  for w in weights if isinstance(weights, list) else [weights]:
88
- attempt_download(w)
89
  ckpt = torch.load(w, map_location=map_location) # load
90
  model.append(ckpt['ema' if ckpt.get('ema') else 'model'].float().fuse().eval()) # FP32 model
91
 
 
85
  # Loads an ensemble of models weights=[a,b,c] or a single model weights=[a] or weights=a
86
  model = Ensemble()
87
  for w in weights if isinstance(weights, list) else [weights]:
88
+ #attempt_download(w)
89
  ckpt = torch.load(w, map_location=map_location) # load
90
  model.append(ckpt['ema' if ckpt.get('ema') else 'model'].float().fuse().eval()) # FP32 model
91