Update yolov5/models/experimental.py

yolov5/models/experimental.py

@@ -82,10 +82,10 @@ def attempt_load(weights, device=None, inplace=True, fuse=True):
     model = Ensemble()
 
     for w in weights if isinstance(weights, list) else [weights]:
-        with torch.serialization.safe_globals([DetectionModel, Sequential, Conv]):
+        with torch.serialization.safe_globals([DetectionModel, Sequential, Conv, torch.nn.modules.conv.Conv2d]):
             ckpt = torch.load(attempt_download(w), map_location='cpu')  # safe load
         ckpt = (ckpt.get('ema') or ckpt['model']).to(device).float()  # FP32 model
-
+    
         # Model compatibility updates
         if not hasattr(ckpt, 'stride'):
             ckpt.stride = torch.tensor([32.])