Solved by ptrblck in post #2 
              
              
                Your forward method accepts an x input, while you are using input as the first argument to your conv layer. 
Change it to self.conv1(x) and it should work.