Fix: downgrade the torchversion to solve the yolov5 model inplace problem

TODO
- Load the yolov5 model and rexport to higher version's torch with inplace setting for Harwish

requirements.txt

@@ -6,8 +6,8 @@ PyYAML
 requests
 scipy
 toml
-torch==1.9.0
-torchvision==0.10.0
+torch==1.6.0
+torchvision==0.7.0
 tqdm
 typed-ast
 typing-extensions

src/digitizer/yolov5/models/common.py

@@ -7,7 +7,7 @@ import torch.nn.functional as F
 from torch import Tensor
 
 # Avoid the use of original Hardswish at torch to avoid the error:
-class Hardswish(nn.Module):
+class HardswishReplace(nn.Module):
     r"""Applies the hardswish function, element-wise, as described in the paper:
 
     `Searching for MobileNetV3`_.
@@ -66,7 +66,7 @@ class Conv(nn.Module):
         super(Conv, self).__init__()
         self.conv = nn.Conv2d(c1, c2, k, s, autopad(k, p), groups=g, bias=False)
         self.bn = nn.BatchNorm2d(c2)
-        self.act = Hardswish() if act else nn.Identity()
+        self.act = nn.Hardswish() if act else nn.Identity()
 
     def forward(self, x):
         return self.act(self.bn(self.conv(x)))