fix test

lanpa · May 8, 2018 · 00232a0 · 00232a0
1 parent f76e92b
commit 00232a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo_graph.py b/demo_graph.py
@@ -81,7 +81,7 @@ def forward(self, x):
 
 class SimpleModel(nn.Module):
     def __init__(self):
-        super().__init__()
+        super(SimpleModel, self).__init__()
     def forward(self, x):
         return x*2