Update DummyModel.py

DummyModel.py

@@ -7,4 +7,7 @@ class DummyModel(torch.nn.Module):
         super().__init__()
 
     def forward(self, x):
-        return 1
+        return 1
+
+    def __call__(self, x):
+        return self.forward(x)