Update networks.py

networks.py

@@ -128,7 +128,8 @@ class FeatureRegression(nn.Module):
 
     def forward(self, x):
         x = self.conv(x)
-        x = x.view(x.size(0), -1)
+        # Change view() to reshape() and make contiguous
+        x = x.contiguous().view(x.size(0), -1)
         x = self.linear(x)
         x = self.tanh(x)
         return x