Merge pull request #191 from baoxin1100/master

fix fuse vgg bug
ppogg · Aug 5, 2022 · d6c8038 · d6c8038
2 parents 529b4b4 + f3bf53c
commit d6c8038
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models/yolo.py b/models/yolo.py
@@ -204,9 +204,9 @@ def fuse(self):  # fuse model Conv2d() + BatchNorm2d() layers
                     m.rbr_dense = rbr_reparam
                     # m.__delattr__('rbr_dense')
                     m.__delattr__('rbr_1x1')
-                    if hasattr(self, 'rbr_identity'):
+                    if hasattr(m, 'rbr_identity'):
                         m.__delattr__('rbr_identity')
-                    if hasattr(self, 'id_tensor'):
+                    if hasattr(m, 'id_tensor'):
                         m.__delattr__('id_tensor')
                     m.deploy = True
                     delattr(m, 'se')