Create new file

DummyModel.py

@@ -0,0 +1,10 @@
+import torch
+import torch.nn
+
+
+class DummyModel(torch.nn.Module):
+    def __init__(self):
+        super().__init__()
+
+    def foward(self, x):
+        return 1