Solved by albanD in post #4 
              
              
                Yes, squeeze() is dangerous in this case. You can use .view(-1, 6144) if you want as well (where the -1 will adapt to your current batch size). 
The error comes from the loss here that does not handle the misshared input/labels. Not sure why the error is not caught earlier though.