Update model.py

model.py

@@ -17,4 +17,8 @@ class Model(nn.Module):
         x = self.elu2(self.bn2(self.ll2(x)))
         x = torch.sum(x, dim=1)
         x = self.llf(x)
-        return x
+        return x
+
+
+if __name__ == '__main__':
+    model = torch.load('model.pth')