Browse Source

实现AlexNet白盒水印集成

liyan 1 năm trước cách đây
mục cha
commit
456cc5166c
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  1. 1 1
      model/Alexnet.py

+ 1 - 1
model/Alexnet.py

@@ -54,7 +54,7 @@ class Alexnet(nn.Module):
         
     def forward(self, x):
         x = self.features(x)
-        x = x.view(x.size(0), -1)
+        x = x.reshape(x.size(0), -1)
         x = self.classifier(x)
         return x