Solved by ptrblck in post #7 
              
              
                This shape mismatch error might be raised by a linear layer, which expects another in_features shape. 
I’m currently unsure which layer it would be so you would need to check for in_features=7680 in your model definition and change it to 3072, as this seems to be the number of features used in the a…