From 055e72af5b887832d5e7267ac9226c825d498cd2 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 10 Mar 2022 12:58:41 +0100
Subject: [PATCH] Optimize PyTorch 1.11.0 compatibility update (#6933)

---
 models/experimental.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/models/experimental.py b/models/experimental.py
index 01bdfe72db4f..782ecbeface9 100644
--- a/models/experimental.py
+++ b/models/experimental.py
@@ -106,10 +106,10 @@ def attempt_load(weights, map_location=None, inplace=True, fuse=True):
                 if not isinstance(m.anchor_grid, list):  # new Detect Layer compatibility
                     delattr(m, 'anchor_grid')
                     setattr(m, 'anchor_grid', [torch.zeros(1)] * m.nl)
-        elif t is nn.Upsample:
-            m.recompute_scale_factor = None  # torch 1.11.0 compatibility
         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
 
     if len(model) == 1:
         return model[-1]  # return model