Suppress torch 1.9.0 max_pool2d() warning (#4227)

ultralytics · Jul 29, 2021 · 18f6ba7 · 18f6ba7
1 parent c2c958c
commit 18f6ba7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/models/common.py b/models/common.py
@@ -1,6 +1,7 @@
 # YOLOv5 common modules
 
 import logging
+import warnings
 from copy import copy
 from pathlib import Path
 
@@ -158,7 +159,9 @@ def __init__(self, c1, c2, k=(5, 9, 13)):
 
     def forward(self, x):
         x = self.cv1(x)
-        return self.cv2(torch.cat([x] + [m(x) for m in self.m], 1))
+        with warnings.catch_warnings():
+            warnings.simplefilter('ignore')  # suppress torch 1.9.0 max_pool2d() warning
+            return self.cv2(torch.cat([x] + [m(x) for m in self.m], 1))
 
 
 class Focus(nn.Module):