--- def forward(x):  # should return a tuple
+++ def forward(self, x):  # should return a tuple