[Before posting a query, check the FAQs - it might already be answered!](Google Colab)
Your linear layer is raising the shape mismatch so make sure the input activation to this layer matches the in_features.
PS: you can post code snippets by wrapping them into three backticks ```, which makes debugging easier.