mirror
/
MODNet
mirror of https://github.com/ZHKKKe/MODNet.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add torchscript

pull/76/head
kevin 2021-03-02 17:47:36 +08:00
parent 26033da8f6
commit 7309a1ed79
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/models/backbones/mobilenetv2.py
View File

@ -136,7 +136,7 @@ class MobileNetV2(nn.Module):
# Initialize weights # Initialize weights
self._init_weights() self._init_weights()
def forward(self, x, feature_names=None): def forward(self, x):
# Stage1 # Stage1
x = self.features[0](x) x = self.features[0](x)
x = self.features[1](x) x = self.features[1](x)