Update badnet_m.py
Browse files- badnet_m.py +1 -1
badnet_m.py
CHANGED
|
@@ -25,7 +25,7 @@ class BadNet(nn.Module):
|
|
| 25 |
x = self.conv2(x)
|
| 26 |
x = F.relu(x)
|
| 27 |
x = self.pool(x)
|
| 28 |
-
x = x.view(-1, self.fc_features)
|
| 29 |
x = self.fc1(x)
|
| 30 |
x = F.relu(x)
|
| 31 |
x = self.fc2(x)
|
|
|
|
| 25 |
x = self.conv2(x)
|
| 26 |
x = F.relu(x)
|
| 27 |
x = self.pool(x)
|
| 28 |
+
x = x.contiguous().view(-1, self.fc_features)
|
| 29 |
x = self.fc1(x)
|
| 30 |
x = F.relu(x)
|
| 31 |
x = self.fc2(x)
|