Merge pull request deepcam-cn#157 from ilgrad/master

torch 1.11.0 compatibility, Upsample
zhn6818 · Apr 8, 2022 · f5dda40 · f5dda40
2 parents 89b0175 + dd1f7d5
commit f5dda40
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/models/yolo.py b/models/yolo.py
@@ -228,6 +228,8 @@ def fuse(self):  # fuse model Conv2d() + BatchNorm2d() layers
                 m.conv = fuse_conv_and_bn(m.conv, m.bn)  # update conv
                 delattr(m, 'bn')  # remove batchnorm
                 m.forward = m.fuseforward  # update forward
+            elif type(m) is nn.Upsample:
+                m.recompute_scale_factor = None  # torch 1.11.0 compatibility
         self.info()
         return self