Create train.py

train.py

@@ -0,0 +1,19 @@
+import torch
+import torch.nn as nn
+
+class SensorEncoder(nn.Module):
+    def __init__(self):
+        super().__init__()
+        self.fc = nn.Linear(10, 32)
+
+    def forward(self, x):
+        return self.fc(x)
+
+def main():
+    model = SensorEncoder()
+    dummy_input = torch.randn(1, 10)
+    output = model(dummy_input)
+    print(output)
+
+if __name__ == "__main__":
+    main()