Well, the issue is I am not sure if its working. I have a keras program with same number of layers and other hyperparameters, it gives 92% accuracy. But the pytorch model gives 20% accuracy on the same data.  Can you please explain what is the difference between return x and return self.fc.forward(x) in the forward function.