Solved by ptrblck in post #2 
              
              
                You are accidentally returning the self.fc2 layer in your model: 
x = self.fc2
return x

instead of calling it with x and returning the activation. 
Change it to: 
x = self.fc2(x)
return x

and it should work.