Upload model

model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c4956d1399d3b3712d17c57d45b400a2f0bba42669fcc4fb9634eb4d52986e16
+size 1548

model.py

@@ -0,0 +1,24 @@
+import torch 
+import torch.nn as nun
+
+class SimpleModel(nun.Module): 
+    def __init__(self):
+        super(SimpleModel, self).__init__()
+        self.linear = nun.Linear(10, 1)
+    def forward(self, x): 
+        return self.linear(x)
+
+
+model= SimpleModel()
+model.linear
+
+x = torch.randn(1, 10)
+t1 = x.to(torch.float)
+
+with torch.no_grad():
+    prediction = model(t1).tolist()
+    
+print(prediction)
+
+model= SimpleModel()
+torch.save(model.state_dict(),'model.pth')