Solved by Dipam_Vasani in post #2 
              
              
                Linear layer consists of matrix multiplication. For matrix multiplication, the number of columns of matrix 1 should be equal to number of rows of matrix 2. In your case the first one has [10 * 3584] while the second one has [25088 * 10] hence you get a mismatch. The 3584 and 25088 should match. I wo…