Deprecate torch 1.6.0 compat _non_persistent_buffers_set (#8797)

Deprecate torch 1.6.0 compat _non_persistent_buffers_set
ultralytics · Jul 30, 2022 · 7921351 · 7921351
1 parent ec4de43
commit 7921351
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/models/experimental.py b/models/experimental.py
@@ -89,8 +89,6 @@ def attempt_load(weights, device=None, inplace=True, fuse=True):
             if t is Detect and not isinstance(m.anchor_grid, list):
                 delattr(m, 'anchor_grid')
                 setattr(m, 'anchor_grid', [torch.zeros(1)] * m.nl)
-        elif t is Conv:
-            m._non_persistent_buffers_set = set()  # torch 1.6.0 compatibility
         elif t is nn.Upsample and not hasattr(m, 'recompute_scale_factor'):
             m.recompute_scale_factor = None  # torch 1.11.0 compatibility