update model

Author: JavaZeroo

model.py

@@ -28,4 +28,4 @@ def forward(self, x):
         out = self.sigmoid(out)
         out = self.fc2(out)
         out = self.sigmoid(out)
-        return out
+        return out.mul(x)